button bug fix