diff --git a/python/samples/02-agents/providers/azure/openai_client_with_session.py b/python/samples/02-agents/providers/azure/openai_client_with_session.py
index 4aceecb5ab..addf89329f 100644
--- a/python/samples/02-agents/providers/azure/openai_client_with_session.py
+++ b/python/samples/02-agents/providers/azure/openai_client_with_session.py
@@ -4,7 +4,7 @@ import asyncio
 from random import randint
 from typing import Annotated
 
-from agent_framework import Agent, AgentSession, ChatOptions, tool
+from agent_framework import Agent, AgentSession, tool
 from agent_framework.openai import OpenAIChatClient
 from azure.identity import AzureCliCredential
 from dotenv import load_dotenv
@@ -76,19 +76,19 @@ async def example_with_session_persistence_in_memory() -> None:
     # First conversation
     query1 = "What's the weather like in Tokyo?"
     print(f"User: {query1}")
-    result1 = await agent.run(query1, session=session, options=ChatOptions(store=False))
+    result1 = await agent.run(query1, session=session, options={"store": False})
     print(f"Agent: {result1.text}")
 
     # Second conversation using the same session - maintains context
     query2 = "How about London?"
     print(f"\nUser: {query2}")
-    result2 = await agent.run(query2, session=session, options=ChatOptions(store=False))
+    result2 = await agent.run(query2, session=session, options={"store": False})
     print(f"Agent: {result2.text}")
 
     # Third conversation - agent should remember both previous cities
     query3 = "Which of the cities I asked about has better weather?"
     print(f"\nUser: {query3}")
-    result3 = await agent.run(query3, session=session, options=ChatOptions(store=False))
+    result3 = await agent.run(query3, session=session, options={"store": False})
     print(f"Agent: {result3.text}")
     print("Note: The agent remembers context from previous messages in the same session.\n")