Al-Bara' b. 'Azib reported that this verse:"Allah keeps those who believe steadfast with firm word in this world and the Hereafter" was revealed in connection with the torment of the grave

English translation · Sahih Muslim