trying to fix button issue