diff --git a/cypress/integration/1-store/setElements.spec.ts b/cypress/integration/1-store/setElements.spec.ts
index 75fc2ee9..dd302070 100644
--- a/cypress/integration/1-store/setElements.spec.ts
+++ b/cypress/integration/1-store/setElements.spec.ts
@@ -1,5 +1,5 @@
 import { getElements } from '../../../examples/Stress/utils'
-import { useStore } from '~/composables'
+import { useVueFlow } from '~/composables'
 import { Edge, FlowStore, Node } from '~/types'
 import { isEdge, isNode } from '~/utils'
 
@@ -17,7 +17,7 @@ describe('test store action setElements', () => {
   }
 
   it('sets elements', async () => {
-    const store = useStore()
+    const { store } = useVueFlow()
     await setElements(store)
     expect(store.nodes).to.have.length(2)
     expect(store.edges).to.have.length(1)
@@ -26,7 +26,7 @@ describe('test store action setElements', () => {
   context('elements pre-set', () => {
     let store: FlowStore
     beforeEach(async () => {
-      store = useStore()
+      const { store } = useVueFlow()
       await setElements(store)
     })
 
diff --git a/cypress/integration/1-store/setState.spec.ts b/cypress/integration/1-store/setState.spec.ts
index 09b04f47..12cb395b 100644
--- a/cypress/integration/1-store/setState.spec.ts
+++ b/cypress/integration/1-store/setState.spec.ts
@@ -1,10 +1,10 @@
-import { useStore } from '~/composables'
+import { useVueFlow } from '~/composables'
 import { FlowState, FlowStore } from '~/types'
 import { initialState } from '~/store'
 describe('test store state', () => {
   let store: FlowStore
 
-  beforeEach(() => (store = useStore()))
+  beforeEach(() => ({ store } = useVueFlow()))
 
   it('has any initial state', () => {
     expect(store.state).to.exist
@@ -27,7 +27,7 @@ describe('test store state', () => {
   })
 
   it('takes initial options', () => {
-    const store = useStore({
+    const { store } = useVueFlow({
       zoomOnScroll: false,
     })
     expect(store.zoomOnScroll).to.eq(false)