diff --git a/Jenkinsfile-create-dataset-docker b/Jenkinsfile-create-dataset-docker
index 871c33b..4d4d0f9 100644
--- a/Jenkinsfile-create-dataset-docker
+++ b/Jenkinsfile-create-dataset-docker
@@ -1,5 +1,7 @@
 pipeline {
-    agent any
+     agent {
+         dockerfile true
+     }
     parameters{
      string(
          defaultValue: 'piotrwrzodak',
@@ -36,17 +38,10 @@ pipeline {
                    sh 'kaggle datasets download -d thedevastator/airbnb-prices-in-european-cities'
                    sh 'unzip airbnb-prices-in-european-cities.zip'
                    sh 'ls'
+                   sh 'python3 create-dataset.py'
+                   archiveArtifacts artifacts: 'data/barcelona_weekends.train.csv, data/barcelona_weekends.dev.csv, data/barcelona_weekends.test.csv', fingerprint: true
                 }
             }
         }
-        stage('Docker') {
-            agent {
-                dockerfile true
-            }
-            steps {
-                sh 'python3 create-dataset.py'
-                archiveArtifacts artifacts: 'data/barcelona_weekends.train.csv, data/barcelona_weekends.dev.csv, data/barcelona_weekends.test.csv', fingerprint: true
-            }
-        }
    }
 }
\ No newline at end of file