diff --git a/Dockerfile b/Dockerfile
index 2602a7f..463b914 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -9,6 +9,7 @@ RUN pip3 install kaggle
 RUN pip3 install tensorflow
 RUN pip3 install sklearn
 RUN pip3 install matplotlib
+RUN pip3 install mlflow
 COPY ./data_dev ./
 COPY ./evaluate_network.py ./
 RUN mkdir /.kaggle
diff --git a/JenkinsfileNeural b/JenkinsfileNeural
index c09ecfe..7417d67 100644
--- a/JenkinsfileNeural
+++ b/JenkinsfileNeural
@@ -16,7 +16,7 @@ node {
 
         }
         stage('Clone repo') {
-                   /* try {*/ docker.image("karopa/ium:28").inside {
+                   /* try {*/ docker.image("karopa/ium:29").inside {
                         stage('Test') {
                             checkout([$class: 'GitSCM', branches: [[name: '*/mlflow']], doGenerateSubmoduleConfigurations: false, extensions: [], submoduleCfg: [], userRemoteConfigs: [[url: 'https://git.wmi.amu.edu.pl/s434765/ium_434765']]])
                             copyArtifacts fingerprintArtifacts: true, projectName: 's434765-create-dataset', selector: buildParameter("BUILD_SELECTOR")