diff --git a/template/steps/{% if hyperparameters_tuning %}hp_tuning{% endif %}/hp_tuning_single_search.py b/template/steps/{% if hyperparameters_tuning %}hp_tuning{% endif %}/hp_tuning_single_search.py
index 0a5b022..261ab71 100644
--- a/template/steps/{% if hyperparameters_tuning %}hp_tuning{% endif %}/hp_tuning_single_search.py	
+++ b/template/steps/{% if hyperparameters_tuning %}hp_tuning{% endif %}/hp_tuning_single_search.py	
@@ -82,6 +82,7 @@ def hp_tuning_single_search(
     log_metadata(
         metadata={"metric": float(score)},
         artifact_name="hp_result",
+        infer_artifact=True,
     )
     ### YOUR CODE ENDS HERE ###
     return cv.best_estimator_