edit button fix