fix edit
small fix