refactor(module): Correctly export new model class