diff --git a/docs/introduction/foundations/components/entity/entity-guide.mdx b/docs/introduction/foundations/components/entity/entity-guide.mdx
index 6a8a1006cd..aa093df706 100644
--- a/docs/introduction/foundations/components/entity/entity-guide.mdx
+++ b/docs/introduction/foundations/components/entity/entity-guide.mdx
@@ -43,7 +43,7 @@ except NoItemsFound as e:
     )
 
     # Poll until the connection is active
-    connected_account = request.wait_until_active(client=composio_client, timeout=100)
+    connected_account = request.wait_until_active(client=toolset.client, timeout=100)
 ```
 </CodeGroup>
 </Tab>
@@ -396,4 +396,4 @@ curl --request POST \
 </Tab>
 </Tabs>
 
-You can use the above pattern to make agents at a user level to perform tasks on their behalf.
\ No newline at end of file
+You can use the above pattern to make agents at a user level to perform tasks on their behalf.