fix up button bug