New idea to improve the immortality totem

My idea is an improvement for the immortality totems since although they can save you sometimes one wishes they were better, for example you are exploring in the water (or in aquifers in the new versions) and you start to drown and your totem wears out Even if I save you, you will continue dying.My idea is that, for example, if you die from a slow fall, if you die drowned you will be given watery breathing, if you die fighting you will resist and that the effect of resistance to fire is removed from the defect of the totem and only apply if you die in lava