diff --git a/MantleAPI/include/MantleAPI/Traffic/i_entity_repository.h b/MantleAPI/include/MantleAPI/Traffic/i_entity_repository.h
index c909077770f99ef272199a1375c133f62e583f19..ed103f04bd57c4b2d67a8ce519d2c1d9edd9e647 100644
--- a/MantleAPI/include/MantleAPI/Traffic/i_entity_repository.h
+++ b/MantleAPI/include/MantleAPI/Traffic/i_entity_repository.h
@@ -48,6 +48,7 @@ class IEntityRepository
 
     virtual void RegisterEntityCreatedCallback(const std::function<void(IEntity&)>& callback) = 0;
     virtual void RegisterEntityDeletedCallback(const std::function<void(const std::string&)>& callback) = 0;
+    virtual void RegisterEntityDeletedCallback(const std::function<void(UniqueId)>& callback) = 0;
 };
 
 }  // namespace mantle_api