diff --git a/JenkinsfileDatasetStats b/JenkinsfileDatasetStats
index 5f4589d..6f9c4ba 100644
--- a/JenkinsfileDatasetStats
+++ b/JenkinsfileDatasetStats
@@ -1,6 +1,10 @@
 pipeline{
-    agent any
-
+    agent{ 
+    dockerfile {
+                    filename: 'lab4.dockerfile',
+                    reuseNode: true,
+                }
+    }
     parameters{
             buildSelector(
                 defaultSelector: lastSuccessful(),
@@ -18,12 +22,6 @@ pipeline{
         }
         }
         stage('Prepare image and stats') {
-            agent{
-                dockerfile{
-                    filename: 'Lab4.dockerfile',
-                    reuseNode: true,
-                }
-            }
             steps {
                 script {
                     copyArtifacts filter: 'wines_test.csv,wines_dev.csv,wines_train.csv', fingerprintArtifacts: true, projectName: 'z-s458023-create-dataset', selector: workspace()