Ideas about ways to keep aatrox revive

Aatrox is a champion who after his rework had nothing similar to the old expect the revive.I understand that aatrox is in a preety good spot right now but i have an idea to keep the revive without being completely problematic as it is now.

When old aatrox revived after having stacked his w couldnt revive again for some minutes because it was put in a cooldown .only the attack speed would stack from the blood well.I think something similar should happen to this aatrox.If he is killed after getting a kill he will be revived but the revive will be put in a cooldown and cant be revived again after he gets a takedown with his ult on for some minutes.

I just would like to keep at least 1 thing from the old aatrox identity and i think that would be a great way to keep it and make it balanced.I hope riot sees this and does some changes.Feel free to tell your opinions