Use the entity name as a key