fixing button bug