diff --git a/src/cmdforge/gui/dialogs/test_step_dialog.py b/src/cmdforge/gui/dialogs/test_step_dialog.py
index 3738197..f1beae2 100644
--- a/src/cmdforge/gui/dialogs/test_step_dialog.py
+++ b/src/cmdforge/gui/dialogs/test_step_dialog.py
@@ -136,8 +136,12 @@ class TestStepDialog(QDialog):
         top_layout.setContentsMargins(0, 0, 0, 0)
 
         # Left: Variables input
-        vars_group = QGroupBox("Input Variables")
-        vars_group.setToolTip(
+        vars_group = QGroupBox()
+        vars_layout = QVBoxLayout(vars_group)
+
+        vars_title = QLabel("Input Variables")
+        vars_title.setStyleSheet("font-weight: bold; font-size: 13px;")
+        vars_title.setToolTip(
             "<p><b>Input Variables</b></p>"
             "<p>These are the variables referenced by this step using {variable} syntax.</p>"
             "<p>Enter test values to simulate what the step would receive during actual execution.</p>"
@@ -145,7 +149,7 @@ class TestStepDialog(QDialog):
             "<b>{argname}</b> - Values from tool arguments<br>"
             "<b>{prev_output}</b> - Output from previous steps</p>"
         )
-        vars_layout = QVBoxLayout(vars_group)
+        vars_layout.addWidget(vars_title)
 
         vars_help = QLabel("Provide test values for variables used in this step:")
         vars_help.setStyleSheet("color: #718096; font-size: 11px;")
@@ -161,8 +165,12 @@ class TestStepDialog(QDialog):
         top_layout.addWidget(vars_group, 1)
 
         # Right: Assertions
-        assert_group = QGroupBox("Assertions (Optional)")
-        assert_group.setToolTip(
+        assert_group = QGroupBox()
+        assert_layout = QVBoxLayout(assert_group)
+
+        assert_title = QLabel("Assertions (Optional)")
+        assert_title.setStyleSheet("font-weight: bold; font-size: 13px;")
+        assert_title.setToolTip(
             "<p><b>Assertions</b></p>"
             "<p>Define checks to automatically validate the step output.</p>"
             "<p><b>Available assertion types:</b><br>"
@@ -175,7 +183,7 @@ class TestStepDialog(QDialog):
             "• <b>Matches Regex</b> - Output must match a pattern<br>"
             "• <b>Min/Max Length</b> - Output length constraints</p>"
         )
-        assert_layout = QVBoxLayout(assert_group)
+        assert_layout.addWidget(assert_title)
 
         assert_help = QLabel("Define checks to validate the step output:")
         assert_help.setStyleSheet("color: #718096; font-size: 11px;")
@@ -250,8 +258,12 @@ class TestStepDialog(QDialog):
         bottom_layout.addLayout(controls_layout)
 
         # Output section
-        output_group = QGroupBox("Output")
-        output_group.setToolTip(
+        output_group = QGroupBox()
+        output_layout = QVBoxLayout(output_group)
+
+        output_title = QLabel("Output")
+        output_title.setStyleSheet("font-weight: bold; font-size: 13px;")
+        output_title.setToolTip(
             "<p><b>Output</b></p>"
             "<p>Shows the results after running the step:</p>"
             "<p>• <b>Status</b> - Success/failure and execution time<br>"
@@ -259,7 +271,7 @@ class TestStepDialog(QDialog):
             "• <b>Output variables</b> - Variables that would be available to subsequent steps<br>"
             "• <b>Assertion results</b> - Pass/fail status for each assertion</p>"
         )
-        output_layout = QVBoxLayout(output_group)
+        output_layout.addWidget(output_title)
 
         # Status line
         self.status_label = QLabel("Click 'Run Step' to test this step")