My suggestion would be not allowing them to "dispute the punishment", but encouraging them to logically follow the path of their actions to the conclusion that

My suggestion would be not allowing them to "dispute the punishment", but encouraging them to logically follow the path of their actions to the conclusion that the punishment is designed to avoid. You're not interested in teaching them to justify poor judgement, you're trying to teach them the judgement skills directly. Of course, situations vary.