From 14485439934f087a483a0fb152dcde220e3159f9 Mon Sep 17 00:00:00 2001
From: braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Sun, 5 Feb 2023 23:44:57 +0100
Subject: [PATCH] chore(core): cleanup useVueFlow

Signed-off-by: braks <78412429+bcakmakoglu@users.noreply.github.com>
---
 packages/core/src/composables/useVueFlow.ts | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/packages/core/src/composables/useVueFlow.ts b/packages/core/src/composables/useVueFlow.ts
index fd298445..27e30379 100644
--- a/packages/core/src/composables/useVueFlow.ts
+++ b/packages/core/src/composables/useVueFlow.ts
@@ -137,11 +137,15 @@ export default (options?: FlowProps): VueFlowStore => {
     if (isParentScope) {
       // dispose of state values and storage entry
       tryOnScopeDispose(() => {
-        if (storage.get(vueFlow!.id)) {
-          vueFlow!.$destroy()
-        }
+        if (vueFlow) {
+          const storedInstance = storage.get(vueFlow.id)
 
-        vueFlow = null
+          if (storedInstance) {
+            storedInstance.$destroy()
+          } else {
+            warn(`No store instance found for id ${vueFlow.id} in storage.`)
+          }
+        }
       })
     }
   }