Treeview fix