fixed button bug