diff --git a/Jenkinsfile-Docker b/Jenkinsfile-Docker
index 48764e4..631184f 100644
--- a/Jenkinsfile-Docker
+++ b/Jenkinsfile-Docker
@@ -18,14 +18,14 @@ stages {
     stage('Train') {
         steps {
             sh 'ls -l'
-            sh 'docker run wujt python3 train.py 89'
+            sh 'docker run wujt python3 train.py'
         }
     }
     stage('Prediction') {
         steps {
             sh 'ls -l'
-            sh 'docker run wujt python3 pred.py 89'
-            archiveArtifacts artifacts: 'predictionResults1.csv'
+            sh 'docker run wujt python3 pred.py'
+            archiveArtifacts artifacts: 'predictionResults.csv'
         }
     }
 }