diff --git a/Jenkinsfile_mlflow b/Jenkinsfile_mlflow
index 613605e..5c1809f 100644
--- a/Jenkinsfile_mlflow
+++ b/Jenkinsfile_mlflow
@@ -1,7 +1,7 @@
 pipeline {
     agent { 
-        docker {
-            image 'kambobdocker420/ium:mlflow'
+        dockerfile {
+            filename 'Dockerfile'
             args '-v /mlruns:/mlruns'
         }
     }