Fix error with editing custom field