From 35afe3b9816039a117ce5f5f73af561bf3abdd56 Mon Sep 17 00:00:00 2001
From: antonykamp <info@antonykamp.de>
Date: Tue, 18 Apr 2023 14:32:26 +0200
Subject: [PATCH] fix: add return statement

---
 yaramo/node.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/yaramo/node.py b/yaramo/node.py
index b3d74f9..cb57972 100644
--- a/yaramo/node.py
+++ b/yaramo/node.py
@@ -94,7 +94,7 @@ def remove_edge_to_node(self, node: "Node"):
 
     def get_edge_to_node(self, node):
         """Returns the edge to the given neighbor node."""
-        next(edge for edge in self.connected_edges if edge.get_opposite_node(self) == node)
+        return next(edge for edge in self.connected_edges if edge.get_opposite_node(self) == node)
 
     def get_possible_followers(self, source):
         """Returns the Nodes that could follow (head, left, right) when comming from a source Node connected to this Node."""