Clean references to the original User model and use `get_user_model`