diff --git a/packages/core/src/components/Nodes/NodeWrapper.vue b/packages/core/src/components/Nodes/NodeWrapper.vue
index 959d3006..0d90d485 100644
--- a/packages/core/src/components/Nodes/NodeWrapper.vue
+++ b/packages/core/src/components/Nodes/NodeWrapper.vue
@@ -92,16 +92,13 @@ onUpdateNodeInternals((updateIds) => {
   }
 })
 
-watchEffect(
-  (onCleanup) => {
-    if (!node.hidden && nodeElement.value) {
-      props.resizeObserver.observe(nodeElement.value as HTMLDivElement)
+onMounted(() => {
+  props.resizeObserver.observe(nodeElement.value as HTMLDivElement)
+})
 
-      onCleanup(() => props.resizeObserver.unobserve(nodeElement.value as HTMLDivElement))
-    }
-  },
-  { flush: 'post' },
-)
+onBeforeUnmount(() => {
+  props.resizeObserver.unobserve(nodeElement.value as HTMLDivElement)
+})
 
 watch(
   [() => node.type, () => node.sourcePosition, () => node.targetPosition],