From 2a259c1448258f47da9575a67b1338271c8824fe Mon Sep 17 00:00:00 2001
From: Hebi Li <lihebi.com@gmail.com>
Date: Mon, 14 Aug 2023 00:26:46 -0700
Subject: [PATCH] fix: compile error

---
 ui/src/lib/store/canvasSlice.tsx | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/ui/src/lib/store/canvasSlice.tsx b/ui/src/lib/store/canvasSlice.tsx
index 85bc1323..f3959ed6 100644
--- a/ui/src/lib/store/canvasSlice.tsx
+++ b/ui/src/lib/store/canvasSlice.tsx
@@ -375,10 +375,11 @@ export const createCanvasSlice: StateCreator<MyState, [], [], CanvasSlice> = (
   selectedPods: new Set(),
   selectionParent: undefined,
   selectPod: (id, selected) => {
+    const nodesMap = get().getNodesMap();
     set(
       produce((state: MyState) => {
         if (selected) {
-          const p = get().getPod(id)?.parent;
+          const p = nodesMap.get(id)?.parentNode;
           // if you select a node that has a different parent, clear all previous selections
           if (
             state.selectionParent !== undefined &&
@@ -809,7 +810,7 @@ export const createCanvasSlice: StateCreator<MyState, [], [], CanvasSlice> = (
         }
         toLevel = scope.data.level + 1;
         // FIXME: since richNode and codeNode doesn't have height when it's created, we have to pass its height manually in case crash.
-        const nodeHeight = get().getPod(nodeId)?.height || 0;
+        const nodeHeight = nodesMap.get(nodeId)?.height || 0;
         position = getNodePositionInsideScope(
           node,
           scope,