diff --git a/src/content/docs/user-guide/concepts/agents/session-management.mdx b/src/content/docs/user-guide/concepts/agents/session-management.mdx
index 66c54017b..9d2ac73dd 100644
--- a/src/content/docs/user-guide/concepts/agents/session-management.mdx
+++ b/src/content/docs/user-guide/concepts/agents/session-management.mdx
@@ -71,7 +71,7 @@ The conversation, and associated state, is persisted to the underlying storage b
 Multi-agent systems (Graph/Swarm) can also use session management to persist their state:
 
 ```python
-from strands.multiagent import Graph
+from strands.multiagent import GraphBuilder
 from strands.session.file_session_manager import FileSessionManager
 
 # Create agents