diff --git a/Jenkinsfile_training b/Jenkinsfile_training
index 57b5250..dbca904 100644
--- a/Jenkinsfile_training
+++ b/Jenkinsfile_training
@@ -12,6 +12,11 @@ pipeline {
         stage('Get artifacts') {
             steps {
                 copyArtifacts fingerprintArtifacts: true, projectName: 's444507-create-dataset', selector: lastSuccessful()
+            }
+        }
+        stage('Run mlflow script and save artifacts') {
+            steps {
+                sh "python3 lab08_deepLearining_mlflow.py $epoch"
                 archiveArtifacts artifacts: 'CarPrices_pytorch_model.pkl'
                 archiveArtifacts artifacts: 'mlruns/**'
                 archiveArtifacts artifacts: 'my_model/**'
@@ -20,12 +25,6 @@ pipeline {
                 build job: 's444507-evaluation/master/'
             }
         }
-        stage('Run mlflow script and save artifacts') {
-            steps {
-                sh "python3 lab08_deepLearining_mlflow.py $epoch"
-
-            }
-        }
     }
     post {
         always {