diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index 26b73c5..1935c42 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -20,7 +20,7 @@ pipeline {
         }
         stage('Save trained model files') {
 			steps{
-				archiveArtifacts 'linear_regression/**'
+				archiveArtifacts 'linear_regression.h5
 			}
 		}
     }