diff --git a/model_manager/activities/training.py b/model_manager/activities/training.py
index fb14f98..ae9bae0 100644
--- a/model_manager/activities/training.py
+++ b/model_manager/activities/training.py
@@ -338,6 +338,7 @@ class Training(SientiaMonitoring):
             raise ValueError('Report path, train data path, or test data path is not set')
 
         self.info(f'Storing model for {train_params.model_type}', metadata)
+        wrapper._input_example = None
         wrapper.store_model(name=train_params.model_name)
         self.info(f'Logging artifacts for {train_params.model_type}', metadata)
         mlflow.log_artifact(train_result.report_path)