diff --git a/src/calng/base_correction.py b/src/calng/base_correction.py
index 1697f568ce0c1fecea2ecb6db3ed4097725608e6..33f208a639f7069bb5b98fdd52d18b652c664475 100644
--- a/src/calng/base_correction.py
+++ b/src/calng/base_correction.py
@@ -956,12 +956,13 @@ class BaseCorrection(PythonDevice):
                 return
 
             train_id = metadata.getAttribute("timestamp", "tid")
-            cell_table = np.squeeze(data_hash.get(self._cell_table_path))
-            if len(cell_table.shape) == 0 and int(cell_table) == 0:
+            cell_table = data_hash.get(self._cell_table_path)
+            if cell_table.size == 0:
                 self.log_status_warn(
-                    "cellId had 0 dimensions. DAQ may not be sending data."
+                    "Empty cell table, DAQ probably not sending data."
                 )
                 return
+            cell_table = np.squeeze(cell_table)
 
             # no more common reasons to skip input, so go to processing
             if state is State.ON: