diff --git a/lab5/eval/Jenkinsfile b/lab5/eval/Jenkinsfile
index 9fffa8e..9da2b3f 100644
--- a/lab5/eval/Jenkinsfile
+++ b/lab5/eval/Jenkinsfile
@@ -24,7 +24,7 @@ pipeline {
         stage('docker') {
             agent {
                 docker {
-                    image 'kubakonieczny/ium:train-sacred-mlflow'
+                    image 'kubakonieczny/ium:train-sacred-mlflow2'
                 }
             }
             stages {
diff --git a/lab5/train/requirements.txt b/lab5/train/requirements.txt
index 09d4e09..6d7491d 100644
--- a/lab5/train/requirements.txt
+++ b/lab5/train/requirements.txt
@@ -4,3 +4,4 @@ tensorflow
 keras==2.3.1
 sacred
 mlflow
+matplotlib