Musannaf Ibn Abi Shayba 6:1249

Abu Mu'awiyah narrated to us, from al-A'mash, from Sa'd ibn 'Ubaydah, from al-Bara' ibn 'Azib: "Allah keeps firm those who believe, with the firm word, in the worldly life." He said: The firmness in the worldly life is when the two angels come to the man in the grave and say to him: "Who is your Lord?" He says: "My Lord is Allah." They say: "And what is your religion?" He says: "My religion is Islam." They say: "And who is your Prophet?" He says: "My Prophet is Muhammad, may Allah bless him and grant him peace." That is the firmness in the worldly life.