Musnad Abu Ya'la al-Mawsili 117:237

Zuhayr narrated to us, Affan narrated to us, Hammad narrated to us, Thabit narrated to us, from Anas, that the Messenger of Allah (may Allah bless him and grant him peace) entered upon a man from among his companions to visit him, and he had become like a fledgling bird. He said to him, "Did you ask Allah for anything?" He said, "I said: O Allah, whatever You are going to punish me with in the Hereafter, then hasten it for me in this world." The Messenger of Allah (may Allah bless him and grant him peace) said to him, "You do not have the strength for the punishment of Allah. Why did you not say: 'O Allah, give us in this world good and in the Hereafter good, and protect us from the punishment of the Fire'?"