diff --git a/python/agent-framework/sample-agent/agent.py b/python/agent-framework/sample-agent/agent.py
index 8f9fcb7b..500ad3dc 100644
--- a/python/agent-framework/sample-agent/agent.py
+++ b/python/agent-framework/sample-agent/agent.py
@@ -37,7 +37,7 @@
 # <DependencyImports>
 
 # AgentFramework SDK
-from agent_framework import ChatAgent
+from agent_framework import Agent
 from agent_framework.azure import AzureOpenAIChatClient
 
 # Agent Interface
@@ -156,8 +156,8 @@ def _create_chat_client(self):
     def _create_agent(self):
         """Create the AgentFramework agent with initial configuration"""
         try:
-            self.agent = ChatAgent(
-                chat_client=self.chat_client,
+            self.agent = Agent(
+                client=self.chat_client,
                 instructions=self.AGENT_PROMPT,
                 tools=[],
             )