diff --git a/bec_widgets/widgets/plots/plot_base.py b/bec_widgets/widgets/plots/plot_base.py
index 772c9abb..c0b1b60b 100644
--- a/bec_widgets/widgets/plots/plot_base.py
+++ b/bec_widgets/widgets/plots/plot_base.py
@@ -1158,6 +1158,8 @@ class PlotBase(BECWidget, QWidget):
         for item in self.plot_item.items:
             if not isinstance(item, pg.ImageItem) or item.image is None:
                 continue
+            if item.image.ndim > 2:
+                continue
 
             if item.image_transform is not None:
                 inv_transform, _ = item.image_transform.inverted()