diff --git a/ui/gradio_ui.py b/ui/gradio_ui.py
index d74c064..8192055 100644
--- a/ui/gradio_ui.py
+++ b/ui/gradio_ui.py
@@ -75,7 +75,7 @@ class GenerateUI:
                             inputs.append(engine_dropdown)
                             engine_rows = []
                             for i, engine in enumerate(engines):
-                                with gr.Row(visible=(i == 0)) as engine_row:
+                                with gr.Column(visible=(i == 0)) as engine_row:
                                     gr.Markdown(value=f"## {engine.name}")
                                     engine_rows.append(engine_row)
                                     options = engine.get_options()