Fix EntryModelGenerator