diff --git a/mem0/memory/main.py b/mem0/memory/main.py
index 2e073cef..d5744b68 100644
--- a/mem0/memory/main.py
+++ b/mem0/memory/main.py
@@ -41,7 +41,7 @@ class Memory(MemoryBase):
         self.enable_graph = False
 
         if self.version == "v1.1" and self.config.graph_store.config:
-            from mem0.memory.main_graph import MemoryGraph
+            from mem0.memory.graph_memory import MemoryGraph
             self.graph = MemoryGraph(self.config)
             self.enable_graph = True