Not punishing. Purifying.
The purification is done by the blood of Jesus when He caused us to be born again and gave us a new, sinless nature.
When this body of flesh dies, the last attachment to sin is done away with and all that remains is the already purified upon the moment of acting on saving faith believer.