Buttons bug fixes