Add button bug fix