diff --git a/src/calng/JungfrauCorrection.py b/src/calng/JungfrauCorrection.py
index 58d481c9a4aa6d14758c3d21f319b2a27d893e80..e78d671ba8c5919523b0286fc86d9386132c652e 100644
--- a/src/calng/JungfrauCorrection.py
+++ b/src/calng/JungfrauCorrection.py
@@ -307,9 +307,9 @@ class JungfrauCorrection(BaseCorrection):
     @property
     def input_data_shape(self):
         return (
-            self._schema_cache["dataFormat.memoryCells"],
-            self._schema_cache["dataFormat.pixelsY"],
-            self._schema_cache["dataFormat.pixelsX"],
+            self.unsafe_get("dataFormat.memoryCells"),
+            self.unsafe_get("dataFormat.pixelsY"),
+            self.unsafe_get("dataFormat.pixelsX"),
         )
 
     def __init__(self, config):
@@ -352,7 +352,7 @@ class JungfrauCorrection(BaseCorrection):
         buffer_handle, buffer_array = self._shmem_buffer.next_slot()
         self.kernel_runner.correct(self._correction_flag_enabled)
         self.kernel_runner.reshape(
-            output_order=self._schema_cache["dataFormat.outputAxisOrder"],
+            output_order=self.unsafe_get("dataFormat.outputAxisOrder"),
             out=buffer_array,
         )
 
@@ -364,8 +364,8 @@ class JungfrauCorrection(BaseCorrection):
                 preview_cell,
                 preview_pulse,
             ) = utils.pick_frame_index(
-                self._schema_cache["preview.selectionMode"],
-                self._schema_cache["preview.index"],
+                self.unsafe_get("preview.selectionMode"),
+                self.unsafe_get("preview.index"),
                 cell_table,
                 _pretend_pulse_table,
                 warn_func=self.log_status_warn,