At-Tur · 52:47

47. And for those who wronged themselves through idolatry and sins is a punishment before the punishment of the afterlife: in this life through being killed and imprisoned, and in the Transitory Stage through punishment of the grave. However, most of them do not know that, which is why they remain on their disbelief.