add/remove button fix