diff --git a/src/pygama/pargen/AoE_cal.py b/src/pygama/pargen/AoE_cal.py
index edfd1e74f..8a531d2ee 100644
--- a/src/pygama/pargen/AoE_cal.py
+++ b/src/pygama/pargen/AoE_cal.py
@@ -1023,11 +1023,8 @@ def time_correction(
 
                     elif mode == "none":
                         time_dict = {
-                            time: mean
-                            for time, mean in zip(
-                                np.array(self.timecorr_df.index),
-                                np.nanmean(self.timecorr_df["mean"]),
-                            )
+                            time: np.nanmean(self.timecorr_df["mean"])
+                            for time in np.array(self.timecorr_df.index)
                         }
 
                     else: