diff --git a/src/View/Network/GraphWidget.py b/src/View/Network/GraphWidget.py
index febeeb2d058b602116490ca5f3fe064ce70c677c..e3ac1aedacbd33b20f48fc377c9d01afc7798138 100644
--- a/src/View/Network/GraphWidget.py
+++ b/src/View/Network/GraphWidget.py
@@ -521,14 +521,14 @@ class GraphWidget(QGraphicsView):
         for edge in self.graph.edges():
             n1 = next(
                 filter(
-                    lambda n: n.node.name == edge.node1.name,
+                    lambda n: n.node == edge.node1,
                     self.node_items
                 )
             )
 
             n2 = next(
                 filter(
-                    lambda n: n.node.name == edge.node2.name,
+                    lambda n: n.node == edge.node2,
                     self.node_items
                 )
             )