Punishment During Death
There is no such rule told in Deen. However, those on whom the Hujjah has been completed, they face Azab right from the moment of their death. See the following verse: 8: 50 وَلَوْ تَرَىٰ إِذْ يَتَوَفَّى الَّذِينَ كَفَرُوا ۙ الْمَلَائِكَةُ يَضْرِبُونَ وُجُوهَهُمْ وَأَدْبَارَهُمْ وَذُوقُوا عَذَابَ الْحَرِيقِ