The Prophet ﷺ used to seek refuge with Allah from the punishment of the grave and from the trial of the Dajjal. He said, "You will be tested in your graves."
أخبرنا قتيبة قال حدثنا سفيان عن يحيى عن عمرة
أن النبي ﷺ كان يستعيذ بالله من عذاب القبر ومن فتنة الدجال قال وقال إنكم تفتنون في قبوركم