diff --git a/Jenkinsfile b/Jenkinsfile
index e923388..4835a8b 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -21,11 +21,10 @@ pipeline {
         KAGGLE_KEY="$params.KAGGLE_KEY"
         CUTOFF="$params.CUTOFF"
     }
-    agent {
-        dockerfile {
-            additionalBuildArgs "-t ium"
-        }
-    }
+    agent 
+	 { 
+	 dockerfile true
+	 }
 
     stages {
         stage("Check out from version control") {
diff --git a/Jenkinsfile2 b/Jenkinsfile2
index 716ad98..5acf371 100644
--- a/Jenkinsfile2
+++ b/Jenkinsfile2
@@ -1,9 +1,7 @@
 pipeline {
-    agent {
-        dockerfile {
-            additionalBuildArgs "-t ium"
-        }
-    }
+     agent { 
+	 dockerfile true
+	 }
     parameters{
         buildSelector(
             defaultSelector: lastSuccessful(),