Even if one gets punished in this world then that should be considered a blessing compared to the punishments of akhirah.