diff --git a/equipment_main_ui.py b/equipment_main_ui.py
index 5dc4114..c931dee 100644
--- a/equipment_main_ui.py
+++ b/equipment_main_ui.py
@@ -40,6 +40,10 @@ def run():
             continue
         elif overlay.run_mode == 'paused':
             continue
+        elif overlay.run_mode == 'finished':
+            overlay.update_status_label("finished", "0")
+            overlay.run_mode = 'init'
+            continue
         else:
             break
 
@@ -250,6 +254,8 @@ def run():
                 print('FPS {}'.format(1 / (time() - outer_loop_time)))
                 outer_loop_time = time()
                 cv.waitKey(150)
+        overlay.run_mode = 'finished'
+
 
 
 def check_and_move_tile(capture_win, visio, conf, rect, needl, hsv, tresh, mask, point_source, point_dest):