fix button bugs