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