diff --git a/debye_bec/devices/pilatus/pilatus.py b/debye_bec/devices/pilatus/pilatus.py
index a24ab35..7f51e9b 100644
--- a/debye_bec/devices/pilatus/pilatus.py
+++ b/debye_bec/devices/pilatus/pilatus.py
@@ -181,8 +181,8 @@ class Pilatus(PSIDeviceBase, ADBase):
             height = self.image1.array_size.height.get()
             # Geometry correction for the image
             data = np.reshape(value, (height, width))
-            last_image = self.preview.get()
-            if np.array_equal(data, last_image):
+            last_image: DevicePreviewMessage = self.preview.get()
+            if np.array_equal(data, last_image.data):
                 # No update if image is the same, ~2.5ms on 2400x2400 image (6M)
                 logger.debug(
                     f"Pilatus preview image for {self.name} is the same as last one, not updating."