diff --git a/packages/memory-graph/src/components/memory-graph.tsx b/packages/memory-graph/src/components/memory-graph.tsx
index cf35e234..bdae2e8a 100644
--- a/packages/memory-graph/src/components/memory-graph.tsx
+++ b/packages/memory-graph/src/components/memory-graph.tsx
@@ -497,7 +497,7 @@ export function MemoryGraph({
 	const activeVersionChain = useMemo(() => {
 		if (!activeNodeData || activeNodeData.type !== "memory") return null
 		return chainIndex.current.getChain(activeNodeData.id)
-	}, [activeNodeData])
+	}, [activeNodeData, limitedDocuments])
 
 	const isLoading = externalIsLoading