diff --git a/Jenkinsfile-train b/Jenkinsfile-train
index e75dd8c..df90581 100644
--- a/Jenkinsfile-train
+++ b/Jenkinsfile-train
@@ -27,6 +27,11 @@ pipeline {
     }
     
     stages {
+        stage('Get dataset from artifact') {
+            steps {
+                copyArtifacts projectName: 's444409-create-dataset', selector: lastSuccessful(), optional: true
+            }
+        }
         stage('Train model') {
             steps {
                 sh "python train_model.py with 'epochs=${params.EPOCHS}' 'batch_size=${params.BATCHSIZE}'"
@@ -36,7 +41,7 @@ pipeline {
             steps {
                archiveArtifacts artifacts: 'model_out', onlyIfSuccessful: true
                archiveArtifacts artifacts: 'sacred_runs/**', onlyIfSuccessful: true
-               build job: "s444409-evaluation/${params.BRANCH}/"
+               build job: "s444409-evaluation/${params.BRANCH}/", parameters: [string(name: 'BRANCH', value: "${params.BRANCH}")]
             }
         }
     }