Another test improvement for isEditMode