Tree control tristate issue

I am facing an issue with the Tristate not indicating correctly on the Tree control on reload. I have a tree control with multiple parents. on Selection the tristate is indicated correctly but after save and reload on the immediate parent node is checked but all the other parent state is not updated. What cud be the reason fr the same?

Can you provide a snippet of the problematic code? As far as I can see, in our demo all works as expected.