diff --git a/src/ui/test/test.py b/src/ui/test/test.py
index adc2fc4..cbd35b8 100755
--- a/src/ui/test/test.py
+++ b/src/ui/test/test.py
@@ -443,7 +443,7 @@ class Test(Widget):
                 if step.step_type in ("leak_1", "leak_2"):
                     self.leak_step = step
 
-            step_types = [step.type for step in steps]
+            step_types = [step.step_type for step in steps]
             if "leak_1" in step_types and "leak_2" in step_types:
                 leak1_index = step_types.index("leak_1")
                 leak2_index = step_types.index("leak_2")
@@ -454,7 +454,7 @@ class Test(Widget):
             # Insert 'instruction_extra' after the first 'instructions' if not inserted between leaks
             if not inserted_instruction:
                 for i, step in enumerate(steps):
-                    if step.type == "instructions":
+                    if step.step_type == "instructions":
                         steps.insert(i + 1, Step(step_type="instruction_extra", spec={}))
                         inserted_instruction = True
                         break