fix bug from removing button