TreeGrid | lazy load - nodes are collapsed after scroll

When I open nodes in tree grid and then I scroll down and back then all nodes are collapsed. I created small demo with h2 database to check issue: https://github.com/cervenf/vaadin-treegrid-lazyload Is any way to persist opened nodes? Thank you.