fixing button again