fix edit bugs