Fix bug in removing punishments

Signed-off-by: Jonathan Dieter <jdieter@lesbg.com>