diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index 2a35e40..7623324 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -23,6 +23,7 @@ pipeline {
             description: 'Number of epochs',
             name: 'EPOCHS',
             trim: false
+        )
      }
 
    stages {