diff --git a/modelforge/train/training.py b/modelforge/train/training.py
index d896474f..e96822d1 100644
--- a/modelforge/train/training.py
+++ b/modelforge/train/training.py
@@ -47,6 +47,7 @@
 
 import matplotlib
 
+# Use Agg backend for matplotlib, to avoid tkinter issues when using profiler
 matplotlib.use("Agg")
 
 __all__ = [