Replace User with get_user_model()