Fix edit button.