diff --git a/dan/ocr/document/train.py b/dan/ocr/document/train.py
index 4b005ec4bdbdd5512c7c2d37d825df573cb3f6e1..93829fe0f402014699f2f236bc3b0d21817a5884 100644
--- a/dan/ocr/document/train.py
+++ b/dan/ocr/document/train.py
@@ -28,6 +28,14 @@ try:
 except ImportError:
     MLFLOW = False
 
+try:
+    import mlflow
+
+    MLFLOW = True
+    logger.info("MLflow Logging available.")
+except ImportError:
+    MLFLOW = False
+
 logger = logging.getLogger(__name__)