Even though he lost the power of his Rinnegan eye, because he was mentally prepared for sacrifice and prepared for anything that could happen, he was able to accept such a massive change.