diff --git a/src/store/stateStore.ts b/src/store/stateStore.ts
index 7ec031d9..97f8f0c3 100644
--- a/src/store/stateStore.ts
+++ b/src/store/stateStore.ts
@@ -1,5 +1,5 @@
 import microDiff from 'microdiff'
-import { setActivePinia, createPinia, defineStore, StoreDefinition, acceptHMRUpdate } from 'pinia'
+import { setActivePinia, createPinia, defineStore, StoreDefinition, acceptHMRUpdate, getActivePinia } from 'pinia'
 import { FlowState, FlowActions, Elements, FlowGetters, GraphNode, GraphEdge } from '~/types'
 import {
   getConnectedEdges,
@@ -17,7 +17,7 @@ import {
 const pinia = createPinia()
 
 export default (id: string, preloadedState: FlowState) => {
-  setActivePinia(pinia)
+  setActivePinia(getActivePinia() ?? pinia)
 
   const store: StoreDefinition<string, FlowState, FlowGetters, FlowActions> = defineStore({
     id: id ?? 'vue-flow',