Sahih Muslim Book 40, Hadith Number 6864.

Chapter : The dead would be shown his seat in Paradise and hell, and the affirmation of the torment of the grave and seeking refuge from it.

Anas b. Malik reported that Allah’s Apostle (may peace be upon him) said: When the servant is placed in his grave and his friends retrace their steps. The rest of the hadith is the same as transmitted by Qatada.