diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index c47163b..bc791d1 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -34,7 +34,7 @@ pipeline {
                         sh 'chmod +x dlgssdpytorch.py'
                         sh 'python3 ./dlgssdpytorch.py $PARAMETRY'
                         sh 'chmod +x train_mlflow.py'
-                        sh 'python3 ./train_mlflow.py'
+                        sh 'mlflow run .'
                     }
                 }
 			}