Fix edit form tests