diff --git a/src/View/Network/GraphWidget.py b/src/View/Network/GraphWidget.py
index dd9ff531..5b55d764 100644
--- a/src/View/Network/GraphWidget.py
+++ b/src/View/Network/GraphWidget.py
@@ -73,7 +73,7 @@ class NodeItem(QGraphicsItem):
 
     def shape(self):
         path = QPainterPath()
-        if self.graph.parent._reservoir.get_assoc_to_node(self.node) is None:
+        if self.graph.graph._reservoir.get_assoc_to_node(self.node) is None:
             path.addEllipse(-10, -10, 20, 20)
         else:
             path.addRect(-10, -10, 20, 20)