كتاب الجنة وصفة نعيمها وأهلها
53  
The Book of Paradise, its Description, its Bounties and its Inhabitants
(17)
Chapter: The Deceased Is Shown His Place In Paradise Or The Fire; And Confirmation Of The Torment In The Grave - We Seek Refuge With Allah From That
(17)
بَاب عَرْضِ مَقْعَدِ الْمَيِّتِ مِنْ الْجَنَّةِ أَوْ النَّارِ عَلَيْهِ وَإِثْبَاتِ عَذَابِ الْقَبْرِ وَالتَّعَوُّذِ مِنْهُ
Sahih Muslim 2871 a

Al-Bara' b. `Azib reported Allah's Apostle (ﷺ) as saying:

This verse: "Allah grants steadfastness to those who believe with firm word," was revealed in connection with the torment of the grave. It would be said to him: Who is your Lord? And he would say: Allah is my Lord and Muhammad is my Apostle (ﷺ), and that is (what is implied) by the words of Allah, the Exalted: "Allah keeps steadfast those who believe with firm word in this world and in the Hereafter."
حَدَّثَنَا مُحَمَّدُ بْنُ بَشَّارِ بْنِ عُثْمَانَ الْعَبْدِيُّ، حَدَّثَنَا مُحَمَّدُ بْنُ جَعْفَرٍ، حَدَّثَنَا شُعْبَةُ، عَنْ عَلْقَمَةَ بْنِ مَرْثَدٍ، عَنْ سَعْدِ بْنِ عُبَيْدَةَ، عَنِ الْبَرَاءِ بْنِ عَازِبٍ، عَنِ النَّبِيِّ صلى الله عليه وسلم قَالَ ‏"‏ ‏{‏ يُثَبِّتُ اللَّهُ الَّذِينَ آمَنُوا بِالْقَوْلِ الثَّابِتِ‏}‏ قَالَ ‏"‏ نَزَلَتْ فِي عَذَابِ الْقَبْرِ فَيُقَالُ لَهُ مَنْ رَبُّكَ فَيَقُولُ رَبِّيَ اللَّهُ وَنَبِيِّيَ مُحَمَّدٌ صلى الله عليه وسلم ‏.‏ فَذَلِكَ قَوْلُهُ عَزَّ وَجَلَّ ‏{‏ يُثَبِّتُ اللَّهُ الَّذِينَ آمَنُوا بِالْقَوْلِ الثَّابِتِ فِي الْحَيَاةِ الدُّنْيَا وَفِي الآخِرَةِ‏}‏ ‏"‏ ‏.‏
Reference : Sahih Muslim 2871a
In-book reference : Book 53, Hadith 88
USC-MSA web (English) reference : Book 40, Hadith 6865
  (deprecated numbering scheme)