diff --git a/jenkins/Jenkinsfile_docker b/jenkins/Jenkinsfile_docker
index d2c3aaa..ba46440 100644
--- a/jenkins/Jenkinsfile_docker
+++ b/jenkins/Jenkinsfile_docker
@@ -1,14 +1,14 @@
 pipeline {
     // properties([[$class: 'GogsProjectProperty', gogsBranchFilter: '', gogsSecret: <object of type hudson.util.Secret>, gogsUsePayload: false], [$class: 'RebuildSettings', autoRebuild: false, rebuildDisabled: false], parameters([string('CUTOFF')]), pipelineTriggers([pollSCM('')])])
 
-    agent {
-        dockerfile {
-            additionalBuildArgs "--build-arg KAGGLE_USERNAME=${params.KAGGLE_USERNAME} --build-arg KAGGLE_KEY=${params.KAGGLE_KEY} -t s478841-create-dataset"
-        }
-    }
 
     stages {
         stage('sh: Shell script') {
+            agent {
+                dockerfile {
+                    additionalBuildArgs "--build-arg KAGGLE_USERNAME=${params.KAGGLE_USERNAME} --build-arg KAGGLE_KEY=${params.KAGGLE_KEY} -t s478841-create-dataset"
+                }
+            }
             steps {
                 sh 'chmod u+x ./scripts/data_stats.sh'
                 sh './scripts/data_stats.sh'