diff --git a/debug.env b/debug.env
index d11b2f2..2e56bb7 100644
--- a/debug.env
+++ b/debug.env
@@ -2,7 +2,7 @@ PYTHONUNBUFFERED=1
 modal.state.teamId=86
 modal.state.workspaceId=146
 modal.state.length=10
-TASK_ID=281
+TASK_ID=325
 SERVER_ADDRESS="put your value here in secret_debug.env"
 API_TOKEN="put your value here in secret_debug.env"
 AGENT_TOKEN="put your value here in secret_debug.env"
diff --git a/src/my_main_script.py b/src/my_main_script.py
index 40026a4..2bd3856 100644
--- a/src/my_main_script.py
+++ b/src/my_main_script.py
@@ -40,11 +40,17 @@ def main():
             "state": None,
             "context": None,
             "command": "preprocessing",
-        }
+        }#,
+        # {
+        #     "state": None,
+        #     "context": None,
+        #     "command": "stop",
+        # }
     ]
 
     # Run application service
     my_app.run(data=data, state=state, initial_events=initial_events)
+    my_app.wait_all()
 
 
 if __name__ == "__main__":