Fixed bug in treeview