diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py
index 7400502..4572f09 100644
--- a/laborious/utils/repository/model_repository.py
+++ b/laborious/utils/repository/model_repository.py
@@ -441,7 +441,7 @@ class MLFlowRepository:
         """
         self.logger.debug(f'Model {model_name} is outdated, downloading a new one')
 
-        del self.model_cache[model_key]['target']['model']
+        del self.model_cache[model_key]['target']
         del self.model_cache[model_key]
 
     def get_model(self, model_name: str, retention: int, model_type: str, flavor: str) -> Any: