diff --git a/src/components/graph/SelectionOverlay.vue b/src/components/graph/SelectionOverlay.vue
index 9365a3b954..aa1f11c766 100644
--- a/src/components/graph/SelectionOverlay.vue
+++ b/src/components/graph/SelectionOverlay.vue
@@ -57,9 +57,7 @@ whenever(
   { immediate: true }
 )
 
-whenever(() => canvasStore.getCanvas().ds.state, positionSelectionOverlay, {
-  deep: true
-})
+canvasStore.getCanvas().ds.onChanged = positionSelectionOverlay
 
 watch(
   () => canvasStore.canvas?.state?.draggingItems,
diff --git a/src/scripts/app.ts b/src/scripts/app.ts
index 86ea2f8e05..fe5fdbcbe1 100644
--- a/src/scripts/app.ts
+++ b/src/scripts/app.ts
@@ -769,7 +769,6 @@ export class ComfyApp {
     this.canvas = new LGraphCanvas(canvasEl, this.graph)
     // Make canvas states reactive so we can observe changes on them.
     this.canvas.state = reactive(this.canvas.state)
-    this.canvas.ds.state = reactive(this.canvas.ds.state)
 
     // @ts-expect-error fixme ts strict error
     this.ctx = canvasEl.getContext('2d')