diff --git a/python/ribasim/ribasim/model.py b/python/ribasim/ribasim/model.py
index e0c1a2dc7..25121c0ab 100644
--- a/python/ribasim/ribasim/model.py
+++ b/python/ribasim/ribasim/model.py
@@ -260,14 +260,13 @@ def plot(self, ax=None, indicate_subnetworks: bool = True) -> Any:
 
         handles, labels = ax.get_legend_handles_labels()
 
-        # TODO
-        # if indicate_subnetworks:
-        #     (
-        #         handles_subnetworks,
-        #         labels_subnetworks,
-        #     ) = node.plot_allocation_networks(ax=ax, zorder=1)
-        #     handles += handles_subnetworks
-        #     labels += labels_subnetworks
+        if indicate_subnetworks:
+            (
+                handles_subnetworks,
+                labels_subnetworks,
+            ) = node.plot_allocation_networks(ax=ax, zorder=1)
+            handles += handles_subnetworks
+            labels += labels_subnetworks
 
         ax.legend(handles, labels, loc="lower left", bbox_to_anchor=(1, 0.5))