diff --git a/template/inference_config.yaml b/template/inference_config.yaml
index 19b3b37..cf569ba 100644
--- a/template/inference_config.yaml
+++ b/template/inference_config.yaml
@@ -13,7 +13,7 @@ settings:
       - sklearn
       - slack
 extra:
-  mlflow_model_name: e2e_use_case_model
+  mlflow_model_name: {{ product_name }}
 {%- if target_environment == 'production' %}
   target_env: Production
 {%- else %}
diff --git a/template/train_config.yaml b/template/train_config.yaml
index 237947e..ceba161 100644
--- a/template/train_config.yaml
+++ b/template/train_config.yaml
@@ -13,7 +13,7 @@ settings:
       - sklearn
       - slack
 extra:
-  mlflow_model_name: e2e_use_case_model
+  mlflow_model_name: {{ product_name }}
 {%- if target_environment == 'production' %}
   target_env: Production
 {%- else %}