diff --git a/src/ui/test/test.py b/src/ui/test/test.py
index d2c8123..48fdd49 100755
--- a/src/ui/test/test.py
+++ b/src/ui/test/test.py
@@ -382,6 +382,7 @@ class Test(Widget):
                         continue
                     steps.insert(i, Steps(type="done"))
                     print_found = True
+                    self.print_step = step
                     if self.config["hardware_config"].get("enforce_piece_removal", "no") == "yes":
                         steps.append(Steps(type="piece_removal"))
                     if count_found: