Update pwm to match simplification