Fix bug with getting custom user model