mat-tree with mat-tab checkboxes unchecked

I have created mat-tabs dynamically and added checkboxes to the same. The following Stackblitz will show the working sample application. But when I switch over to the second tab and return back to the first tab, the values of the first tab get unchecked(I mean the checked symbols go off). How should I preserve that?

Stackblitz Mat-tree Mat-tab Example