TreeUi: lost nodes fixed when tree gets collapsed on expansion callback