diff --git a/src/anemoi/training/commands/mlflow.py b/src/anemoi/training/commands/mlflow.py
index b8b94f5b..e5e5f6a8 100644
--- a/src/anemoi/training/commands/mlflow.py
+++ b/src/anemoi/training/commands/mlflow.py
@@ -6,8 +6,6 @@
 # nor does it submit to any jurisdiction.
 
 
-from anemoi.training.diagnostics.mlflow.auth import TokenAuth
-
 from . import Command
 
 
@@ -40,6 +38,8 @@ def add_arguments(self, command_parser):
 
     def run(self, args):
         if args.subcommand == "login":
+            from anemoi.training.diagnostics.mlflow.auth import TokenAuth
+
             TokenAuth(url=args.url).login(force_credentials=args.force_credentials)
             return