The Prophet used to seek refuge with Allah from the torment of the grave and the tribulation of the Dajjal, and he said: "You will be tried in your graves."
Sunan an-Nasa'i Book 50, Hadith 77