ON/OFF button fix