diff --git a/scripts/plot_network.py b/scripts/plot_network.py
index 53c52187..3d0641d5 100644
--- a/scripts/plot_network.py
+++ b/scripts/plot_network.py
@@ -310,7 +310,7 @@ def plot_h2_map(network, regions):
 
     else:
 
-        h2_total = h2_new
+        h2_total = h2_new.p_nom_opt
 
     link_widths_total = h2_total / linewidth_factor