Allay with a Totem Of Undying

An Allay with a Totem Of Undying will be useless at first, but it will revive you when you die. All of you items will not be so lucky if you died next to lava. Yes, the Ally has to fly to your place of death and this is because imagine you gave your Allay a Totem Of Undying while you yourself keeps one, so I want to make sure that yours will always go off before the Allay