diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index f07e460..cf6f3a6 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -25,7 +25,7 @@ node {
      checkout([$class: 'GitSCM', branches: [[name: '*/master']], extensions: [], userRemoteConfigs: [[credentialsId: 's487197', url: 'https://git.wmi.amu.edu.pl/s487197/ium_487197']]])
  }
  stage('Dockerfile'){
-    def testImage = docker.image('s487197/ium:55')
+    def testImage = docker.image('s487197/ium:56')
 	testImage.inside{ 
          copyArtifacts filter: 'baltimore_train.csv', projectName: 's487197-create-dataset'
 		 sh "python3 ium_sacred.py -epochs $EPOCHS -lr $LR -validation_split $VALIDATION_SPLIT"
diff --git a/MLproject.txt b/MLproject.txt
index 5bf82a3..36b6943 100644
--- a/MLproject.txt
+++ b/MLproject.txt
@@ -1,7 +1,7 @@
 name: s487197
 
 docker_env:
-  image: s487197/ium:55
+  image: s487197/ium:56
 
 entry_points:
   main: