why did you remove that part?
it's supposed to give back focus to the treeview after the context menu of the treeview is closed.