diff --git a/Jenkinsfile_create_dataset b/Jenkinsfile_create_dataset
index 360d050..fbc91b5 100644
--- a/Jenkinsfile_create_dataset
+++ b/Jenkinsfile_create_dataset
@@ -61,7 +61,7 @@ pipeline {
                   archiveArtifacts 'Y_test.csv'
                   archiveArtifacts 'Y_dev.csv'
                   archiveArtifacts 'Y_train.csv'
-                  archiveArtifacts 'prediction.txt'
+                  archiveArtifacts 'prediction.csv'
          }
       }
       stage('clear_after') {