Reviving the Revive

Not too long ago aatrox had his revive removed, personaly as an aatrox main i prefer it to be total trash as long as it is in his kit. But because pros used it as a diving tool it got removed. My idea is to make it so that the resorce bar fills from any self healing and when you reach a cap of around 2000 hp you then get the revive as long as your ult is still active. Im ok with aatrox getting nerfs to counter balance this change i just want to have this part of aatrox come back.