diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index 6ab4528..2e36c3c 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -21,8 +21,8 @@ pipeline {
 	   }
 	stage('Train model with sacred') {
 	    steps {
-	        sh 'python3 ./biblioteka_DL/dllib.py with "epochs=$EPOCHS"'
-            archiveArtifacts artifacts: 'model.pkl, s444018_sacred_FileObserver/**/*.*, result.csv', followSymlinks: false
+	        sh 'python3 ./biblioteka_DL/dllib.py "epochs=$EPOCHS"'
+            archiveArtifacts artifacts: 'model.pkl, result.csv', followSymlinks: false
             archiveArtifacts artifacts: 'mlruns/**'
 			archiveArtifacts artifacts: 'my_model/**'
             build job: 's444018-evaluation/master/'
diff --git a/biblioteka_DL/dllib.py b/biblioteka_DL/dllib.py
index 5b6ef99..d61a0c9 100644
--- a/biblioteka_DL/dllib.py
+++ b/biblioteka_DL/dllib.py
@@ -15,8 +15,7 @@ from urllib.parse import urlparse
 
 # mlflow.set_tracking_uri("http://172.17.0.1:5000")
 mlflow.set_experiment("s444018")
-
-epochs = int(sys.argv[1]) if len(sys.argv) > 1 else 20
+epochs = sys.argv[1]
 
 
 def drop_relevant_columns(imbd_data):