Display properly when editing