fix button woes