diff --git a/mlair/data_handler/data_handler_single_station.py b/mlair/data_handler/data_handler_single_station.py
index 6190784f27a3257c12cf7f6228d5e10324c61a89..c898ca14eee8fad51371c4889d1c2f81aacf7727 100644
--- a/mlair/data_handler/data_handler_single_station.py
+++ b/mlair/data_handler/data_handler_single_station.py
@@ -586,11 +586,11 @@ class DataHandlerSingleStation(AbstractDataHandler):
             non_nan_history = self.history.dropna(dim=dim)
             non_nan_label = self.label.dropna(dim=dim)
             non_nan_observation = self.observation.dropna(dim=dim)
-            if non_nan_label.shape[2] == 0:
+            if non_nan_label.coords[dim].shape[0] == 0:
                 raise ValueError(f'self.label consist of NaNs only - station {self.station} is therefore dropped')
-            if non_nan_history.shape[2] == 0:
+            if non_nan_history.coords[dim].shape[0] == 0:
                 raise ValueError(f'self.history consist of NaNs only - station {self.station} is therefore dropped')
-            if non_nan_observation.shape[2] == 0:
+            if non_nan_observation.coords[dim].shape[0] == 0:
                 raise ValueError(f'self.observation consist of NaNs only - station {self.station} is therefore dropped')
             intersect = reduce(np.intersect1d, (non_nan_history.coords[dim].values, non_nan_label.coords[dim].values,
                                                 non_nan_observation.coords[dim].values))