diff --git a/src/agent_gsm.py b/src/agent_gsm.py
index 1536a23..aa35c69 100644
--- a/src/agent_gsm.py
+++ b/src/agent_gsm.py
@@ -263,7 +263,8 @@ def perform_answerer():
     
     # Setup analyzer script
     # Run answering script
-    while True:
+    attempt_idx = 0
+    while attempt_idx < CONFIG.TaskLimit or CONFIG.TaskLimit is None or CONFIG.TaskLimit == 0:
         # Remove old recording
         if CONFIG.RecordFile.exists():
             os.remove(CONFIG.RecordFile)