diff --git a/fluents/plots.py b/fluents/plots.py
index 7230e1b..4b4cde5 100644
--- a/fluents/plots.py
+++ b/fluents/plots.py
@@ -663,7 +663,7 @@ class NetworkPlot(Plot):
                                                  c=self._nodecolor,
                                                  linewidth=lw,
                                                  zorder=3)
-        self._mappable = node_collection
+        self._mappable = self.node_collection
         # selected nodes is a transparent graph that adjust node-edge visibility
         # according to the current selection needed to get get the selected
         # nodes 'on top' as zorder may not be defined individually