diff --git a/scripts/cluster_network.py b/scripts/cluster_network.py
index 6a8e526b..ce1e4319 100644
--- a/scripts/cluster_network.py
+++ b/scripts/cluster_network.py
@@ -257,7 +257,7 @@ def busmap_for_n_clusters(n, n_clusters, solver_name, focus_weights=None, algori
         else:
             raise ValueError(f"`algorithm` must be one of 'kmeans', 'spectral' or 'louvain'. Is {algorithm}.")
 
-    return (n.buses.groupby(['country', 'sub_network'], group_keys=False, squeeze=True)
+    return (n.buses.groupby(['country', 'sub_network'], group_keys=False)
             .apply(busmap_for_country).rename('busmap'))
 
 def plot_busmap_for_n_clusters(n, n_clusters=50):