diff --git a/python/samples/agentchat_graphrag/app.py b/python/samples/agentchat_graphrag/app.py
index bd43126744f0..736aed159d47 100644
--- a/python/samples/agentchat_graphrag/app.py
+++ b/python/samples/agentchat_graphrag/app.py
@@ -1,4 +1,5 @@
 import asyncio
+from autogen_agentchat.messages import TextMessage
 from autogen_ext.models.openai import AzureOpenAIChatCompletionClient
 from autogen_ext.tools.graphrag import (
     GlobalSearchTool,
@@ -57,7 +58,7 @@ async def main():
     
     response_stream = team.run_stream(task=query)
     async for msg in response_stream:
-        if hasattr(msg, "content"):
+        if isinstance(msg, TextMessage) and hasattr(msg, "content"):
             print(f"\nAgent response: {msg.content}")