Done for custom User model