diff --git a/ra_aid/agent_backends/ciayn_agent.py b/ra_aid/agent_backends/ciayn_agent.py
index 067e80d..9286653 100644
--- a/ra_aid/agent_backends/ciayn_agent.py
+++ b/ra_aid/agent_backends/ciayn_agent.py
@@ -825,7 +825,8 @@ class CiaynAgent:
             try:
                 last_result = self._execute_tool(response)
                 self.chat_history.append(response)
-                self.fallback_handler.reset_fallback_handler()
+                if hasattr(self.fallback_handler, 'reset_fallback_handler'):
+                    self.fallback_handler.reset_fallback_handler()
                 yield {}
 
             except ToolExecutionError as e:
diff --git a/ra_aid/agent_utils.py b/ra_aid/agent_utils.py
index fa1ff62..0abb164 100644
--- a/ra_aid/agent_utils.py
+++ b/ra_aid/agent_utils.py
@@ -464,7 +464,7 @@ def run_agent_with_retry(
 
                 try:
                     _run_agent_stream(agent, msg_list)
-                    if fallback_handler:
+                    if fallback_handler and hasattr(fallback_handler, 'reset_fallback_handler'):
                         fallback_handler.reset_fallback_handler()
                     should_break, prompt, auto_test, test_attempts = (
                         _execute_test_command_wrapper(