How to go back to previous tree after Shift Path?

Hi everyone, I need a little hand with this data tree issue:
I have an initial tree, which I shift back into 2 branches to perform some operations.
I then need to go back to the initial tree after said operations, or simply after the shift tree. Any ideas? Thanks

GoBackToTree.gh (15.4 KB)

I believe you could just use the unflatten component.

Yup, great! Thank you Micheal!