diff --git a/earth2mip/score_ensemble_outputs.py b/earth2mip/score_ensemble_outputs.py
index 6c3e45b..0593d24 100644
--- a/earth2mip/score_ensemble_outputs.py
+++ b/earth2mip/score_ensemble_outputs.py
@@ -110,7 +110,7 @@ def main(
 
             if time_averaging_window:
                 verification = verification.resample(time=time_averaging_window).mean(
-                    dim="time", keep_attrs=True, skipna=False, keepdims=True
+                    dim="time", keep_attrs=True, skipna=False
                 )
 
             ensemble_mse = (verification - ensemble_mean) ** 2.0