diff --git a/mlair/plotting/data_insight_plotting.py b/mlair/plotting/data_insight_plotting.py
index aca5e3c53eca42b3bcb2266afbb481829eb90a62..45f5d3d6a451200d35c8f9085b31d3c394ae62b1 100644
--- a/mlair/plotting/data_insight_plotting.py
+++ b/mlair/plotting/data_insight_plotting.py
@@ -444,6 +444,7 @@ class PlotAvailabilityHistogram(AbstractPlotClass):  # pragma: no cover
         plt.tight_layout()
 
 
+@TimeTrackingWrapper
 class PlotDataHistogram(AbstractPlotClass):  # pragma: no cover
     """
     Plot histogram on transformed input and target data. This data is the same that the model sees during training. No
@@ -570,6 +571,7 @@ class PlotDataHistogram(AbstractPlotClass):  # pragma: no cover
         plt.close('all')
 
 
+@TimeTrackingWrapper
 class PlotPeriodogram(AbstractPlotClass):  # pragma: no cover
     """
     Create Lomb-Scargle periodogram in raw input and target data. The Lomb-Scargle version can deal with missing values.