diff --git a/Jenkinsfile.train b/Jenkinsfile.train
index 1f4f174..c01a7ff 100644
--- a/Jenkinsfile.train
+++ b/Jenkinsfile.train
@@ -3,7 +3,7 @@ pipeline {
         dockerfile true
     }
     parameters {
-        string(name: 'EPOCH', defaultValue: '10', description: 'Number of epochs in training')
+        string(name: 'EPOCHS', defaultValue: '10', description: 'Number of epochs in training')
     }
     stages {
         stage("Git clone") {
@@ -14,7 +14,7 @@ pipeline {
         stage("Training") {
             steps {
                     sh "chmod 777 ml_training.py"
-                    sh "python3 ml_training.py $EPOCH"
+                    sh "python3 ml_training.py"
             }
         }
     }