Fix bug with edit form