diff --git a/Jenkinsfile b/Jenkinsfile
index 8ac9773..44578e6 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -38,12 +38,14 @@ pipeline {
 
          stage('Train and Predict') {
             steps {
+              script {
                def customImage = docker.build("custom-image")
                     customImage.inside {
                         sh 'python3 ./model.py'
                         sh 'python3 ./prediction.py'
                         archiveArtifacts artifacts: 'model.pth, predictions.txt', onlyIfSuccessful: true
                     }
+              }
             }
         }
      }