diff --git a/Jenkinsfile_mlflow b/Jenkinsfile_mlflow
index 4c9bd29..613605e 100644
--- a/Jenkinsfile_mlflow
+++ b/Jenkinsfile_mlflow
@@ -1,9 +1,12 @@
 pipeline {
     agent { 
-        dockerfile  true
-        args '-v /mlruns:/mlruns'
+        docker {
+            image 'kambobdocker420/ium:mlflow'
+            args '-v /mlruns:/mlruns'
+        }
     }
 
+
     parameters {
         string(
             defaultValue: 'kamilab0bkowska',