Correct bug with edit button