diff --git a/src/codegen/agents/code_agent.py b/src/codegen/agents/code_agent.py
index 99c01a874..e27f485da 100644
--- a/src/codegen/agents/code_agent.py
+++ b/src/codegen/agents/code_agent.py
@@ -120,7 +120,7 @@ def run(self, prompt: str, thread_id: Optional[str] = None) -> str:
             if isinstance(message, tuple):
                 print(message)
             else:
-                if isinstance(message, AIMessage) and isinstance(message.content, list) and "text" in message.content[0]:
+                if isinstance(message, AIMessage) and isinstance(message.content, list) and len(message.content) > 0 and "text" in message.content[0]:
                     AIMessage(message.content[0]["text"]).pretty_print()
                 else:
                     message.pretty_print()