fix js in punishment