diff --git a/aixplain/modules/agent/__init__.py b/aixplain/modules/agent/__init__.py
index e08c47b3..c647a2f7 100644
--- a/aixplain/modules/agent/__init__.py
+++ b/aixplain/modules/agent/__init__.py
@@ -541,6 +541,7 @@ def sync_poll(
                     error_message="No response from the service.",
                 )
                 logging.error(f"Polling for Agent: polling for {name}: {e}")
+                return response_body
                 break
 
         # Display completion message
@@ -640,8 +641,8 @@ def run(
             result = self.sync_poll(
                 poll_url, name=name, timeout=timeout, wait_time=wait_time, progress_verbosity=progress_verbosity
             )
-            # if result.status == ResponseStatus.FAILED:
-            #    raise Exception("Model failed to run with error: " + result.error_message)
+            if result.status == ResponseStatus.FAILED:
+               raise Exception("Model failed to run with error: " + result.error_message)
             result_data = result.get("data") or {}
             return AgentResponse(
                 status=ResponseStatus.SUCCESS,