diff --git a/bun.lock b/bun.lock
index 40d15a9ea..73137d994 100644
--- a/bun.lock
+++ b/bun.lock
@@ -163,7 +163,7 @@
         "@repo/lib": "workspace:*",
         "@repo/validation": "workspace:*",
         "@sentry/nextjs": "^10.33.0",
-        "@supermemory/memory-graph": "^0.2.0",
+        "@supermemory/memory-graph": "^0.2.3",
         "@tailwindcss/typography": "^0.5.16",
         "@tanstack/react-form": "^1.12.4",
         "@tanstack/react-query": "^5.90.14",
@@ -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 7d537b7df..aea9f1d6d 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 17c73f7bb..291c04dd3 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") {