diff --git a/rdagent/utils/agent/ret.py b/rdagent/utils/agent/ret.py
index 9409eee4..5065a8a0 100644
--- a/rdagent/utils/agent/ret.py
+++ b/rdagent/utils/agent/ret.py
@@ -35,6 +35,7 @@ class PythonAgentOut(AgentOut):
         if match:
             code = match.group(1)
             return code
+        return resp
 
 
 class BatchEditOut(AgentOut):