diff --git a/scripts/demos/sensors/cameras.py b/scripts/demos/sensors/cameras.py
index ad2bc8c150d..9cad2545cbb 100644
--- a/scripts/demos/sensors/cameras.py
+++ b/scripts/demos/sensors/cameras.py
@@ -138,7 +138,10 @@ def save_images_grid(
     ncol = int(np.ceil(n_images / nrow))
 
     fig, axes = plt.subplots(nrow, ncol, figsize=(ncol * 2, nrow * 2))
-    axes = axes.flatten()
+    if isinstance(axes, np.ndarray):
+        axes = axes.flatten()
+    else:
+        axes = np.array([axes])
 
     # plot images
     for idx, (img, ax) in enumerate(zip(images, axes)):