fix edit buttons