Fixed the custom user model