From 1401bcae64502b488fca2cbe21e021866c628622 Mon Sep 17 00:00:00 2001
From: moklick <info@moritzklack.com>
Date: Sun, 17 Dec 2023 18:15:31 +0100
Subject: [PATCH] chore(edge-wrapper): cleanup

---
 packages/react/src/components/EdgeWrapper/index.tsx | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/packages/react/src/components/EdgeWrapper/index.tsx b/packages/react/src/components/EdgeWrapper/index.tsx
index e82268ac..1462f45f 100644
--- a/packages/react/src/components/EdgeWrapper/index.tsx
+++ b/packages/react/src/components/EdgeWrapper/index.tsx
@@ -112,10 +112,6 @@ function EdgeWrapper({
   const onEdgeClick = (event: React.MouseEvent<SVGGElement, MouseEvent>): void => {
     const { addSelectedEdges, unselectNodesAndEdges, multiSelectionActive } = store.getState();
 
-    if (!edge) {
-      return;
-    }
-
     if (isSelectable) {
       store.setState({ nodesSelectionActive: false });
 
@@ -160,12 +156,12 @@ function EdgeWrapper({
 
   const onKeyDown = (event: KeyboardEvent) => {
     if (elementSelectionKeys.includes(event.key) && isSelectable) {
-      const { unselectNodesAndEdges, addSelectedEdges, edges } = store.getState();
+      const { unselectNodesAndEdges, addSelectedEdges } = store.getState();
       const unselect = event.key === 'Escape';
 
       if (unselect) {
         edgeRef.current?.blur();
-        unselectNodesAndEdges({ edges: [edges.find((e) => e.id === id)!] });
+        unselectNodesAndEdges({ edges: [edge] });
       } else {
         addSelectedEdges([id]);
       }