How to expand the treegrid after invalidateCache

I have a TreeGrid where I call openAll() in the fetchData's DSCallback to open all the nodes.
I call invalidateCache(), which refreshes the tree. However, the tree is now collapsed. How do I expand it again?