diff --git a/bun.lock b/bun.lock
index 40d15a9e..1d049ae6 100644
--- a/bun.lock
+++ b/bun.lock
@@ -298,7 +298,7 @@
     },
     "packages/memory-graph": {
       "name": "@supermemory/memory-graph",
-      "version": "0.2.2",
+      "version": "0.2.3",
       "dependencies": {
         "d3-force": "^3.0.0",
       },
diff --git a/packages/memory-graph/package.json b/packages/memory-graph/package.json
index 7d537b7d..aea9f1d6 100644
--- a/packages/memory-graph/package.json
+++ b/packages/memory-graph/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@supermemory/memory-graph",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "description": "Interactive graph visualization component for Supermemory - visualize and explore your memory connections",
   "type": "module",
   "main": "./src/index.tsx",
diff --git a/packages/memory-graph/src/canvas/renderer.ts b/packages/memory-graph/src/canvas/renderer.ts
index 17c73f7b..291c04dd 100644
--- a/packages/memory-graph/src/canvas/renderer.ts
+++ b/packages/memory-graph/src/canvas/renderer.ts
@@ -244,10 +244,8 @@ function drawEdges(
 			continue
 		}
 
-		const src =
-			typeof edge.source === "string" ? nodeMap.get(edge.source) : edge.source
-		const tgt =
-			typeof edge.target === "string" ? nodeMap.get(edge.target) : edge.target
+		const src = nodeMap.get(srcId)
+		const tgt = nodeMap.get(tgtId)
 		if (!src || !tgt) continue
 
 		if (edgeType === "derives") {