Fix control buttons