fixing button issue