add button bug fix