fix button issue