Small fix in edit view