diff --git a/common/models/role.js b/common/models/role.js index 05638863..e78dc4ed 100644 --- a/common/models/role.js +++ b/common/models/role.js @@ -24,9 +24,9 @@ module.exports = function(Role) { Role.resolveRelatedModels = function() { if (!this.userModel) { var reg = this.registry; - this.roleMappingModel = reg.getModelByType(loopback.RoleMapping); - this.userModel = reg.getModelByType(loopback.User); - this.applicationModel = reg.getModelByType(loopback.Application); + this.roleMappingModel = reg.getModelByType('RoleMapping'); + this.userModel = reg.getModelByType('User'); + this.applicationModel = reg.getModelByType('Application'); } }; diff --git a/test/role.test.js b/test/role.test.js index d6eccf79..c9390c89 100644 --- a/test/role.test.js +++ b/test/role.test.js @@ -47,9 +47,6 @@ describe('role model', function() { ACL.roleMappingModel = RoleMapping; ACL.userModel = User; ACL.applicationModel = Application; - Role.roleMappingModel = RoleMapping; - Role.userModel = User; - Role.applicationModel = Application; }); it('should define role/role relations', function(done) {