diff --git a/mlair/run_modules/post_processing.py b/mlair/run_modules/post_processing.py
index 5691624fbee3302b5815f6b60ef8c3091d42136a..297f378dae13167d2ef9ccac79a86c3038a35718 100644
--- a/mlair/run_modules/post_processing.py
+++ b/mlair/run_modules/post_processing.py
@@ -158,9 +158,10 @@ class PostProcessing(RunEnvironment):
             ds = ds.rename({'XTIME': self.index_dim})
         except ValueError as e:
             logging.warning("Dimension `XTIME' does not exist")
+        ds = ds.squeeze()
         logging.info(f"PostProcessing.load_upstream_wind_sector({name_of_set}: shape of `upstream_wind_sector' is "
                      f"{ds.shape}\ndims are {ds.dims}")
-        self.upstream_wind_sector = ds.squeeze()
+        self.upstream_wind_sector = ds
 
     @TimeTrackingWrapper
     def estimate_sample_uncertainty(self, separate_ahead=False):