diff --git a/Jenkinsfile b/Jenkinsfile
index f5feaad..24ec09d 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -33,7 +33,7 @@ pipeline {
                 {
                        sh 'python3 train.py ${BATCH_SIZE} ${EPOCHS}'
                        sh 'rm -r my_model'
-                       sh 'python3 vgsales-mlflow.py ${BATCH_SIZE} ${EPOCHS}'
+                       sh 'python3 vgsales-mlflow.py'
                 }
             }
 
diff --git a/vgsales-mlflow.py b/vgsales-mlflow.py
index 57add00..81716fb 100644
--- a/vgsales-mlflow.py
+++ b/vgsales-mlflow.py
@@ -66,4 +66,3 @@ with mlflow.start_run():
     mlflow.log_metric("rmse", rmse) 
     #mlflow.keras.log_model(model, 'vgsales_model.h5')
     mlflow.keras.save_model(model, "my_model", signature=mlflow.models.signature.infer_signature(x_train, y_train), input_example=x_train)
-    
\ No newline at end of file