Another edit button fix