fix edit button