fixed small button bug