Fix set on/off button.