Fix edit button