fixing the button issue