diff --git a/Jenkinsfile b/Jenkinsfile
index 5c8f644..f90cc49 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -36,14 +36,16 @@ pipeline {
 
         stage('Download dataset and preprocess data') {
             steps {
-                docker.image('create-dataset-s464863').withRun(
-                "-e KAGGLE_USERNAME=${params.KAGGLE_USERNAME}",
-                "-e KAGGLE_KEY=${params.KAGGLE_KEY}",
-                "-e CUTOFF=${params.CUTOFF}"
-                ) {
-                    sh "chmod +x ./download_dataset.py"
-                    sh "python3 ./download_dataset.py ${params.CUTOFF}"
-                    archiveArtifacts artifacts: 'datasets/*', onlyIfSuccessful: true
+                script {
+                    docker.image('create-dataset-s464863').withRun(
+                    "-e KAGGLE_USERNAME=${params.KAGGLE_USERNAME}",
+                    "-e KAGGLE_KEY=${params.KAGGLE_KEY}",
+                    "-e CUTOFF=${params.CUTOFF}"
+                    ) {
+                        sh "chmod +x ./download_dataset.py"
+                        sh "python3 ./download_dataset.py ${params.CUTOFF}"
+                        archiveArtifacts artifacts: 'datasets/*', onlyIfSuccessful: true
+                    }
                 }
             }
         }