"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.
Sahih Muslim Book 53, Hadith 89