Register models in plugin correctly