diff --git a/Jenkinsfile b/Jenkinsfile
index 28fdc0d..4582214 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -1,16 +1,12 @@
 pipeline {
-    
+    agent {
+        dockerfile true
+    }
     stages{
         stage('Copy data') {
         steps {
          copyArtifacts(projectName: 's452662-create-dataset', selector: lastSuccessful())
         }
-    }
-    stage('Run'){
-        steps {
-        agent {
-        dockerfile true
-    } }
     }
         stage('Save model') {
             steps{