fixing edit button