diff --git a/mlair/data_handler/data_handler_single_station.py b/mlair/data_handler/data_handler_single_station.py
index 038851d05f8b4b3ceeabc7e405acc858affeb874..4217583d4b7ae03a2529deaae38fd33234bba5db 100644
--- a/mlair/data_handler/data_handler_single_station.py
+++ b/mlair/data_handler/data_handler_single_station.py
@@ -163,7 +163,7 @@ class DataHandlerSingleStation(AbstractDataHandler):
         return self.label.squeeze([self.iter_dim, self.target_dim]).transpose(self.time_dim, self.window_dim).copy()
 
     def get_X(self, **kwargs):
-        return self.get_transposed_history()
+        return self.get_transposed_history().sel({self.target_dim: self.variables})
 
     def get_Y(self, **kwargs):
         return self.get_transposed_label()