If you are going back to the Mortis Arc: Maybe the Life and Death Episode from TotJ could be integrated here via visions when Ahsoka gets posessed?

Thematically, it seems like it is worth a shot with Ahsoka getting into a bad situation, closing in on death and then making it out alive through the force…
Also, it would be a nice callback to Ahsokas visions in the bonus Episode.

As it stands, the TotJ Episode is cute and all, but does not have much of a purpose or impact. Integrated here it might actually hit harder.