diff --git a/Jenkinsfile.train b/Jenkinsfile.train
index d8778dd..420d805 100644
--- a/Jenkinsfile.train
+++ b/Jenkinsfile.train
@@ -1,6 +1,6 @@
 pipeline {
     agent {
-		docker { image 'kamilguttmann/ium:train' }
+		dockerfile true
 	}
     parameters {
         string (
@@ -28,7 +28,7 @@ pipeline {
         stage("Train model") {
             steps {
                 sh "chmod u+x ./train_model.py"
-                sh "python3 ./train_model.py with 'epcohs=$EPOCHS'"
+                sh "python3 ./train_model.py with 'epochs=$EPOCHS'"
                 archiveArtifacts artifacts: "model/*, out.csv, experiments/*/*", onlyIfSuccessful: true
             }
         }