Fix to treeview