Fix button bug.