diff --git a/Jenkins_train b/Jenkins_train
index 6c6f0b5..08a85f6 100644
--- a/Jenkins_train
+++ b/Jenkins_train
@@ -13,7 +13,7 @@ pipeline {
         stage('Docker and Training') {
             steps {
                 script {
-                    def dockerImage = docker.image('s487194/ium:1')
+                    def dockerImage = docker.image('s487194/ium:lab5_7')
                     dockerImage.inside {
                         sh "python train.py --epochs ${params.EPOCHS} --lr ${params.LR}"
                         archiveArtifacts artifacts: 'classificationn_model.pt', onlyIfSuccessful: true