diff --git a/python/src/agents/document_agent.py b/python/src/agents/document_agent.py
index 9e9c5fbfc7..597bb69c77 100644
--- a/python/src/agents/document_agent.py
+++ b/python/src/agents/document_agent.py
@@ -76,7 +76,7 @@ def _create_agent(self, **kwargs) -> Agent:
         agent = Agent(
             model=self.model,
             deps_type=DocumentDependencies,
-            result_type=DocumentOperation,
+            output_type=DocumentOperation,
             system_prompt="""You are a Document Management Assistant that helps users create, update, and modify project documents through conversation.
 
 **Your Capabilities:**