diff --git a/Jenkinsfile b/Jenkinsfile
index 87efea4..271b6d1 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -1,4 +1,7 @@
 pipeline {
+    agent {
+        docker { image 'mikolajk/ium:latest' }
+    }
     parameters{
         string(
             defaultValue: 'mikolaj2', 
@@ -16,7 +19,6 @@ pipeline {
             name: 'KAGGLE_KEY'
         )
     }
-    agent { dockerfile true }
     options {
         copyArtifactPermission('s444417-training');
     }
diff --git a/Jenkinsfile3 b/Jenkinsfile3
index 8d74fca..a41fc62 100644
--- a/Jenkinsfile3
+++ b/Jenkinsfile3
@@ -1,7 +1,5 @@
 pipeline {
-    agent {
-        docker { image 'mikolajk/ium:latest' }
-    }
+    agent { dockerfile true }
     stages {
         stage('Test') {
             steps {