Removing Hero button after death

Removing Hero button after death

In the left side of the screen there are the buttons that appear when you get heroes. If I want to make these buttons dissappear after a hero dies, rather than have them be faded out as if awaiting the revival of the hero, how do I do that? When a hero dies, I want to remove the button on the left side of the screen accordingly.

you could do it by removing him, but you can't revive after. I think this'll work,

Event: Unit dies
Conditions: Triggering unit equal to hero is true
Actions: Remove Triggering unit from game

Something like this should work. If you want to revive it later on you need to store it in a variable