diff --git a/docs/data-science/consume-data-agent-python.md b/docs/data-science/consume-data-agent-python.md
index bed37d56df..2ed42e2f91 100644
--- a/docs/data-science/consume-data-agent-python.md
+++ b/docs/data-science/consume-data-agent-python.md
@@ -95,14 +95,14 @@ Use the `InteractiveBrowserCredential` class to authenticate with Microsoft Entr
 
 ```python
 from azure.identity import InteractiveBrowserCredential
-from fabric_data_agent_client import DataAgentClient
+from fabric_data_agent_client import FabricDataAgentClient
 credential = InteractiveBrowserCredential()
 ```
 
 ## Create the data agent client
 
 ```python
-client = DataAgentClient(credential=credential)
+client = FabricDataAgentClient(credential=credential)
 ```
 > [!NOTE]
 > - The `fabric-data-agent-client` package provides the client SDK for connecting to the Fabric data agent.
@@ -149,4 +149,4 @@ This output helps you understand how the agent produced its response and gives t
 
 - [Fabric data agent concepts](concept-data-agent.md)
 - [Fabric data agent SDK](fabric-data-agent-sdk.md)
-- [Data agent end-to-end tutorial](data-agent-end-to-end-tutorial.md)
\ No newline at end of file
+- [Data agent end-to-end tutorial](data-agent-end-to-end-tutorial.md)