fix button bug