diff --git a/xcp_d/interfaces/connectivity.py b/xcp_d/interfaces/connectivity.py
index bb4bf2527..4b7769510 100644
--- a/xcp_d/interfaces/connectivity.py
+++ b/xcp_d/interfaces/connectivity.py
@@ -438,6 +438,9 @@ def _run_interface(self, runtime):
             ax_idx = list(range(ncols))
 
         fig, axes = plt.subplots(nrows=nrows, ncols=ncols, figsize=figsize)
+        if isinstance(axes, plt.Axes):
+            axes = np.array([axes])
+
         for i_ax, atlas in enumerate(selected_atlases):
             atlas_idx = self.inputs.atlases.index(atlas)
             atlas_file = self.inputs.correlations_tsv[atlas_idx]