fix edit feature