That is exactly how it should work and not nearly enough people recognize that. It’s part of the social contract: You break the rules, you get punished. After the punishment, you are a full member of society again.

Otherwise you just get punished again and that should not be the case.