diff --git a/stats/Jenkinsfile b/stats/Jenkinsfile
index 999305c..3da4402 100644
--- a/stats/Jenkinsfile
+++ b/stats/Jenkinsfile
@@ -1,24 +1,9 @@
-pipeline {
-  agent {
-      docker { 
-        image 'jvncziii/ium:latest'
-      }
-    }
-  parameters {
-      buildSelector(
-        defaultSelector: lastSuccessful(),
-        description: 'Which build to use for copying artifacts',
-        name: 'BUILD_SELECTOR'
-      )
-    }
-  stages {
-      stage('Dataset statistics') {
-          steps {
-            copyArtifacts filter:'train.csv', fingerprintArtifacts: true, projectName: 'z-s495716-create-dataset', selector: buildParameter('BUILD_SELECTOR')
-            sh 'chmod 777 stats/data_stats.sh'
-            sh './stats/data_stats.sh'
-            archiveArtifacts artifacts: 'train_lines.txt', followSymlinks: false
-          }
-        }
+ node {
+    checkout scm
+    docker.image('jvncziii/ium:latest').withRun('') { c->
+        copyArtifacts filter:'train.csv', fingerprintArtifacts: true, projectName: 'z-s495716-create-dataset', selector: buildParameter('BUILD_SELECTOR')
+        sh 'chmod 777 stats/data_stats.sh'
+        sh './stats/data_stats.sh'
+        archiveArtifacts artifacts: 'train_lines.txt', followSymlinks: false
       }
   }