diff --git a/mlair/plotting/training_monitoring.py b/mlair/plotting/training_monitoring.py
index 913c11dd8a4e0d23c2bde6864c12f17c65922644..09f49c848675eb21bd172e40b09b265e47c443fb 100644
--- a/mlair/plotting/training_monitoring.py
+++ b/mlair/plotting/training_monitoring.py
@@ -12,7 +12,7 @@ import pandas as pd
 
 from mlair.model_modules.keras_extensions import LearningRateDecay
 
-matplotlib.use('Agg')
+# matplotlib.use('Agg')
 history_object = Union[Dict, keras.callbacks.History]
 lr_object = Union[Dict, LearningRateDecay]