diff --git a/Jenkins_train b/Jenkins_train
index 2f5725e..29e8d7e 100644
--- a/Jenkins_train
+++ b/Jenkins_train
@@ -15,7 +15,7 @@ pipeline {
                 script {
                     def dockerImage = docker.image('s487194/ium:test2')
                     dockerImage.inside { 
-                        sh "python train1.py -epochs ${params.EPOCHS} -lr ${params.LR}"
+                        sh "python train.py -epochs ${params.EPOCHS} -lr ${params.LR}"
                         archiveArtifacts artifacts: 'classificationn_model.pt', onlyIfSuccessful: true
                     }
                 }