Added punishment fix.