diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index 37705f1..9d0b7a9 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -1,7 +1,8 @@
 pipeline {
   	agent {
-		docker { image 'korneliag/ium478815:1.0' }
-	}
+		dockerfile {
+			 additionalBuildArgs "--build-arg KAGGLE_USERNAME=${params.KAGGLE_USERNAME} --build-arg KAGGLE_KEY=51854c15661583860a60db232a7026f0 --build-arg CUTOFF=${params.CUTOFF} -t kgirejko"
+		}
 
      parameters {
 		string(