diff --git a/Dockerfile b/Dockerfile
index 2af5a66..070ca87 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -13,7 +13,7 @@ RUN pip3 install matplotlib
 RUN pip3 install torch
 RUN pip3 install sacred
 RUN pip3 install pymongo
-RUN pip3 install mflow
+RUN pip3 install mlflow
 
 ARG CUTOFF
 ARG KAGGLE_USERNAME
diff --git a/Jenkinsfile_registry b/Jenkinsfile_registry
index 8eec354..7373578 100644
--- a/Jenkinsfile_registry
+++ b/Jenkinsfile_registry
@@ -1,8 +1,9 @@
 pipeline {
     agent {
-	dockerfile {
-			 additionalBuildArgs "-t docker_image"
-		}
+	docker {
+       		image 'docker_image'
+       		args '-v /mlruns:/mlruns'
+	}
     }
 
     stages {
diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index 30fb9f9..6ab4528 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -1,8 +1,9 @@
 pipeline {
   	agent {
-  	    dockerfile {
-			 additionalBuildArgs "--build-arg -v /mlruns:/mlruns -t docker_image"
-		    }
+  	    docker {
+		    image 'docker_image'
+			args '-v /mlruns:/mlruns'
+        	}
 	    }
 	parameters {
     string(