diff --git a/Jenkinsfile_mlflow b/Jenkinsfile_mlflow
index 922ee0b..fb51521 100644
--- a/Jenkinsfile_mlflow
+++ b/Jenkinsfile_mlflow
@@ -31,6 +31,7 @@ pipeline {
                     sh 'python3 ./nn_train_mlflow.py'
                     archiveArtifacts artifacts: 'mlruns/**'
                     sh 'rm -r mlruns'
+                    sh 'rm -r my_model_mlflow'
                 }
             }
         }
diff --git a/nn_train_mlflow.py b/nn_train_mlflow.py
index 912add3..33839bc 100644
--- a/nn_train_mlflow.py
+++ b/nn_train_mlflow.py
@@ -83,6 +83,6 @@ with mlflow.start_run():
     mlflow.log_param("1st_activation_funct", first_activation_funct)
     mlflow.log_param("2nd_activation_funct", second_activation_funct)
     #mlflow.keras.log_model(model, 'my_model')
-    mlflow.keras.save_model(model, "my_model")
+    mlflow.keras.save_model(model, "my_model_mlflow")
     mlflow.log_metric("accuracy", accuracy_score(y_true, y_pred))
     
\ No newline at end of file