Abdullah bin Yasar al-Juhani (may Allah have mercy on him) relates: A man from among us died who had a stomach illness, so we hastened (his burial). Then I came to the mosque and saw that Sayyiduna Sulayman bin Surad and Khalid bin Urfutah (may Allah be pleased with them both) were there. Sulayman said: What prevented you from informing us (of the death) of your companion? I said: He was suffering from a stomach illness, so we hastened (his burial). Sulayman bin Surad (may Allah be pleased with him) said: I heard the Messenger of Allah (peace and blessings be upon him) say: "The one with a stomach illness is not punished in the grave." (Then he asked Khalid:) O Khalid, do you not also bear witness to it? He said: Certainly.
Hadith Referenceالمعجم الكبير للطبراني / باب الخاء / 4092