I think I found a bug with canceling the "entity dies" trigger(please help)

I am making a custom totem that steals an item from the attackers mainhand and gives it to you. It works fine until I die of a non-entity cause, I have treid numerous things to prevent me from dying from indirect causes(like checking if source entity is alive or If they are the damage source) but none have worked. Is this a bug? Let me know if it is or isn't, and if you can, please help. Here is my code https://imgur.com/a/GRzwHTy.