diff --git a/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py b/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py
index b694b123adc4471e64f1ed215c8677be94a83400..716b8bd932210c375dfe729d57cb56903d0d4c6f 100644
--- a/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py
+++ b/rdstar/offline_analysis/fitting/dedicated_fit_approach/plotting_tools.py
@@ -607,7 +607,11 @@ class SystematicsShapePlotter:
             )
 
             axis_index_pair: Tuple[int, int] = comp_to_axis_map[component.name]
-            this_axis: AxesType = axes[axis_index_pair[0], axis_index_pair[1]]
+
+            if len(axes.shape) == 2:
+                this_axis: AxesType = axes[axis_index_pair[0], axis_index_pair[1]]
+            else:
+                this_axis = axes[axis_index_pair[1]]
 
             self._plot_shape_overview(
                 ax=this_axis,