diff --git a/e2e/cypress/component/1-store/edges/removeSelectedEdges.cy.ts b/e2e/cypress/component/1-store/edges/removeSelectedEdges.cy.ts
index 7d366bba..c24e6837 100644
--- a/e2e/cypress/component/1-store/edges/removeSelectedEdges.cy.ts
+++ b/e2e/cypress/component/1-store/edges/removeSelectedEdges.cy.ts
@@ -23,12 +23,12 @@ describe('Store Action: `removeSelectedEdges`', () => {
   })
 
   it('removes selected edges from store', () => {
-    expect(store.getSelectedEdges.value).to.have.length(randomNumber2)
+    expect(store.getSelectedEdges.value).to.have.length(randomNumber - randomNumber2)
   })
 
   it('removes `selected` class from edges', () => {
     cy.get('.vue-flow__edge').then((edges) => {
-      expect(edges[randomNumber2 - 1]).to.have.class('selected')
+      expect(edges[randomNumber2 - 1]).to.not.have.class('selected')
     })
   })
 })
diff --git a/e2e/cypress/component/1-store/nodes/removeSelectedNodes.cy.ts b/e2e/cypress/component/1-store/nodes/removeSelectedNodes.cy.ts
index 870220c3..9677068a 100644
--- a/e2e/cypress/component/1-store/nodes/removeSelectedNodes.cy.ts
+++ b/e2e/cypress/component/1-store/nodes/removeSelectedNodes.cy.ts
@@ -23,12 +23,12 @@ describe('Store Action: `removeSelectedNodes`', () => {
   })
 
   it('removes selected nodes from store', () => {
-    expect(store.getSelectedNodes.value).to.have.length(randomNumber2)
+    expect(store.getSelectedNodes.value).to.have.length(randomNumber - randomNumber2)
   })
 
   it('removes `selected` class from nodes', () => {
     cy.get('.vue-flow__node').then((nodes) => {
-      expect(nodes[randomNumber2 - 1]).to.have.class('selected')
+      expect(nodes[randomNumber2 - 1]).to.not.have.class('selected')
     })
   })
 })