Fix button bug