Small tweak in the treeview