the Prophet used to seek refuge with Allah from the torment of the grave and the trial of the Dajjal, and he said: "You will be tested in your graves."
Sunan an-Nasa'i Book 21, Hadith 248