Make flying pets disperse when die + other ideas

The idea is that when you die currently the flying pets freeze in place when you die

like this

what i think would be cool is having those pets fly outwards

like this, or another idea is make them fly upwards, or even just make them shrink and disappear
then when you respawn they come back
i feel something like this would be cool