Trying to fix edit form bug