diff --git a/src/surfer_h_cli/surferh.py b/src/surfer_h_cli/surferh.py
index 711f59d..e6e9f39 100644
--- a/src/surfer_h_cli/surferh.py
+++ b/src/surfer_h_cli/surferh.py
@@ -316,6 +316,7 @@ def agent_loop(
         url=url,
         screenshots=[browser.screenshot()],
     )
+    set_current_state(current_state)
 
     start_time = time.time()