Add missing entities and repos