diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index eb9bdbf..30fb9f9 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -1,9 +1,8 @@
 pipeline {
   	agent {
-		docker {
-		    image 'docker_image'
-			args '-v /mlruns:/mlruns'
-        	}
+  	    dockerfile {
+			 additionalBuildArgs "--build-arg -v /mlruns:/mlruns -t docker_image"
+		    }
 	    }
 	parameters {
     string(