diff --git a/src/calng/base_correction.py b/src/calng/base_correction.py
index ac50550e7d3ae744d631945fe68db4b03c574485..1f4135bc64e7aab1ef8fdd78ea20b725157bf307 100644
--- a/src/calng/base_correction.py
+++ b/src/calng/base_correction.py
@@ -974,19 +974,13 @@ class BaseCorrection(PythonDevice):
 
             correction_cell_num = self._schema_cache["constantParameters.memoryCells"]
             cell_table_max = np.max(cell_table)
-            if cell_table_max >= correction_cell_num:
-                self.log_status_info(
-                    f"Max cell ID ({cell_table_max}) exceeds range for loaded "
-                    f"constants ({correction_cell_num} cells). Some frames will not be "
-                    "corrected."
-                )
 
             image_data = data_hash.get(self._image_data_path)
-            if image_data.shape[0] != self._schema_cache["dataFormat.memoryCells"]:
+            if cell_table.size != self._schema_cache["dataFormat.memoryCells"]:
                 self.log_status_info(
                     f"Updating new input shape {image_data.shape}, updating buffers"
                 )
-                self.set("dataFormat.memoryCells", image_data.shape[0])
+                self.set("dataFormat.memoryCells", cell_table.size)
                 with self._buffer_lock:
                     self._update_frame_filter()