diff --git a/scripts_python/l2t_train_model.py b/scripts_python/l2t_train_model.py
index 184f28a6..fcd98c2f 100755
--- a/scripts_python/l2t_train_model.py
+++ b/scripts_python/l2t_train_model.py
@@ -11,6 +11,7 @@
 # comet_ml not used, but comet_ml requires to be imported before torch.
 # See bug report here https://github.com/Lightning-AI/lightning/issues/5829
 # Importing now to solve issues later.
+import comet_ml
 import torch
 
 from scilpy.io.utils import assert_inputs_exist, assert_outputs_exist