diff --git a/Jenkinsfile_training b/Jenkinsfile_training
index 8ce5486..d3b2e6d 100644
--- a/Jenkinsfile_training
+++ b/Jenkinsfile_training
@@ -24,7 +24,7 @@ pipeline {
 		}
 		stage('Archive artifacts') {
 			steps{
-				archiveArtifacts artifacts: 'model_movies/**/*.*'
+				archiveArtifacts artifacts: 'model_movies.h5'
 			}
 		}
     }
diff --git a/ium_zadanie6_training.py b/ium_zadanie6_training.py
index 39092a0..0eb28ce 100644
--- a/ium_zadanie6_training.py
+++ b/ium_zadanie6_training.py
@@ -28,6 +28,6 @@ model.fit(
     y = y_train,
     verbose=0, epochs=99)
 	
-model.save('model_movies')
+model.save('model_movies.h5')