diff --git a/src/container/VueFlow/VueFlow.vue b/src/container/VueFlow/VueFlow.vue
index d16c745a..3cad1375 100644
--- a/src/container/VueFlow/VueFlow.vue
+++ b/src/container/VueFlow/VueFlow.vue
@@ -114,27 +114,8 @@ if (store.elements.length) elements.value = store.elements
 
 const options = Object.assign({}, store.$state, props)
 
-const init = async (state: FlowState) => {
-  // set state variables
-  const skip = ['modelValue', 'd3Zoom', 'd3Selection', 'd3ZoomHandler', 'minZoom', 'maxZoom', 'translateExtent', 'nodeExtent']
-  for (const opt of Object.keys(state)) {
-    const val = state[opt as keyof FlowState]
-    if (typeof val !== 'undefined' && !skip.includes(opt)) {
-      if (typeof val === 'object' && !Array.isArray(val)) {
-        ;(store as any)[opt] = { ...(store as any)[opt], ...val }
-      } else (store as any)[opt] = val
-    }
-  }
-  if (!store.isReady) await until(() => store.d3Zoom).not.toBeUndefined()
-  store.setMinZoom(state.minZoom)
-  store.setMaxZoom(state.maxZoom)
-  store.setTranslateExtent(state.translateExtent)
-  store.setNodeExtent(state.nodeExtent)
-}
-onBeforeUnmount(() => store?.$dispose())
-
 invoke(async () => {
-  init(options)
+  store.setState(options)
   store.setElements(elements.value)
   store.isReady = true
 
@@ -158,7 +139,7 @@ invoke(async () => {
   store.instance = instance
   watch(
     () => props,
-    () => init({ ...store.$state, ...props } as FlowState),
+    () => store.setState({ ...store.$state, ...props } as FlowState),
     { flush: 'post', deep: true },
   )
 })
@@ -186,6 +167,8 @@ const transitionName = computed(() => {
   }
   return name
 })
+
+onBeforeUnmount(() => store?.$dispose())
 </script>
 <template>
   <div class="vue-flow">
diff --git a/src/store/stateStore.ts b/src/store/stateStore.ts
index 50fbe9bf..6bfd5b26 100644
--- a/src/store/stateStore.ts
+++ b/src/store/stateStore.ts
@@ -178,6 +178,32 @@ export default (id: string, preloadedState: FlowState) => {
         const { nodes, edges } = parseElements(elements, this.getNodes, this.getEdges, this.nodeExtent)
         this.elements = [...this.elements, ...nodes, ...edges]
       },
+      async setState(state) {
+        // set state variables
+        const skip = [
+          'modelValue',
+          'd3Zoom',
+          'd3Selection',
+          'd3ZoomHandler',
+          'minZoom',
+          'maxZoom',
+          'translateExtent',
+          'nodeExtent',
+        ]
+        for (const opt of Object.keys(state)) {
+          const val = state[opt as keyof FlowState]
+          if (typeof val !== 'undefined' && !skip.includes(opt)) {
+            if (typeof val === 'object' && !Array.isArray(val)) {
+              ;(store as any)[opt] = { ...(store as any)[opt], ...val }
+            } else (store as any)[opt] = val
+          }
+        }
+        if (!this.isReady) await until(() => this.d3Zoom).not.toBeUndefined()
+        this.setMinZoom(state.minZoom)
+        this.setMaxZoom(state.maxZoom)
+        this.setTranslateExtent(state.translateExtent)
+        this.setNodeExtent(state.nodeExtent)
+      },
     },
   })
 
diff --git a/src/types/store.ts b/src/types/store.ts
index 288e6f4b..e231c5b6 100644
--- a/src/types/store.ts
+++ b/src/types/store.ts
@@ -88,7 +88,7 @@ export interface FlowActions {
   updateSize: (size: Dimensions) => void
   setConnectionNodeId: (payload: SetConnectionId) => void
   setInteractive: (isInteractive: boolean) => void
-  addElements: (elements: Elements) => Promise<void>
+  addElements: (elements: Elements) => void
 }
 
 export interface FlowGetters {