diff --git a/mlair/data_handler/data_handler_kz_filter.py b/mlair/data_handler/data_handler_kz_filter.py
index a5a9de6701a03bedad95f54fc22fbd97ee041c86..ce96a8f5c039b5a232aa56765209927dd4019168 100644
--- a/mlair/data_handler/data_handler_kz_filter.py
+++ b/mlair/data_handler/data_handler_kz_filter.py
@@ -41,12 +41,13 @@ class DataHandlerKzFilterSingleStation(DataHandlerSingleStation):
         self.load_data()
         self.interpolate(dim=self.time_dim, method=self.interpolation_method, limit=self.interpolation_limit)
         self.set_inputs_and_targets()
-        import matplotlib
-        matplotlib.use("TkAgg")
-        import matplotlib.pyplot as plt
         self.apply_kz_filter()
-        # self.data.sel(filter="74d", variables="temp", Stations="DEBW107").plot()
-        # self.data.sel(variables="temp", Stations="DEBW107").plot.line(hue="filter")
+        # this is just a code snippet to check the results of the kz filter
+        # import matplotlib
+        # matplotlib.use("TkAgg")
+        # import matplotlib.pyplot as plt
+        # self.input_data.data.sel(filter="74d", variables="temp", Stations="DEBW107").plot()
+        # self.input_data.data.sel(variables="temp", Stations="DEBW107").plot.line(hue="filter")
         if self.do_transformation is True:
             self.call_transform()
         self.make_samples()