diff --git a/src/calng/base_correction.py b/src/calng/base_correction.py
index 33f208a639f7069bb5b98fdd52d18b652c664475..3873d34cc3f2f0b9463e5d2844367f95cc10fc23 100644
--- a/src/calng/base_correction.py
+++ b/src/calng/base_correction.py
@@ -957,7 +957,10 @@ class BaseCorrection(PythonDevice):
 
             train_id = metadata.getAttribute("timestamp", "tid")
             cell_table = data_hash.get(self._cell_table_path)
-            if cell_table.size == 0:
+            if (
+                    (isinstance(cell_table, np.ndarray) and cell_table.size == 0)
+                    or len(cell_table) == 0
+            ):
                 self.log_status_warn(
                     "Empty cell table, DAQ probably not sending data."
                 )