diff --git a/mlair/run_modules/post_processing.py b/mlair/run_modules/post_processing.py
index b4af7a754335e8da6d29870b1a0c4152d7dc9af5..de43f30d929db1de12681d92c9c585df5c07944e 100644
--- a/mlair/run_modules/post_processing.py
+++ b/mlair/run_modules/post_processing.py
@@ -74,7 +74,7 @@ class PostProcessing(RunEnvironment):
         self.plot_path: str = self.data_store.get("plot_path")
         self.target_var = self.data_store.get("target_var")
         self._sampling = self.data_store.get("sampling")
-        self.window_lead_time = extract_value(self.data_store.get("shape_outputs", "model"))
+        self.window_lead_time = extract_value(self.data_store.get("output_shape", "model"))
         self.skill_scores = None
         self.bootstrap_skill_scores = None
         self._run()