From cd0658534a858e82789e42f18d00626b530bddd0 Mon Sep 17 00:00:00 2001
From: braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Wed, 18 Jan 2023 23:29:06 +0100
Subject: [PATCH] examples: cleanup save-restore controls

Signed-off-by: braks <78412429+bcakmakoglu@users.noreply.github.com>
---
 examples/vite/src/SaveRestore/Controls.vue | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/examples/vite/src/SaveRestore/Controls.vue b/examples/vite/src/SaveRestore/Controls.vue
index 5f4d6a2d..85499104 100644
--- a/examples/vite/src/SaveRestore/Controls.vue
+++ b/examples/vite/src/SaveRestore/Controls.vue
@@ -1,6 +1,6 @@
 <script lang="ts" setup>
 import type { FlowExportObject, Node } from '@vue-flow/core'
-import { useVueFlow, useZoomPanHelper } from '@vue-flow/core'
+import { useVueFlow } from '@vue-flow/core'
 
 const flowKey = 'example-flow'
 const state = useStorage<FlowExportObject>(flowKey, {
@@ -12,8 +12,7 @@ const state = useStorage<FlowExportObject>(flowKey, {
 
 const getNodeId = () => `randomnode_${+new Date()}`
 
-const { setTransform } = useZoomPanHelper()
-const { nodes, edges, addNodes, setNodes, setEdges, toObject, dimensions } = useVueFlow()
+const { addNodes, setNodes, setEdges, toObject, dimensions, setTransform } = useVueFlow()
 
 const onSave = () => {
   state.value = toObject()
@@ -24,8 +23,11 @@ const onRestore = () => {
 
   if (flow) {
     const [x = 0, y = 0] = flow.position
+
     setNodes(state.value.nodes)
+
     setEdges(state.value.edges)
+
     setTransform({ x, y, zoom: flow.zoom || 0 })
   }
 }