Aye immunity during DfA seems a good step for it.
Could they not potentially make it a true 2 part attack? First press launches you into the air (with some kind of slow fall giving you a half second or so of air time for part 2) - 2nd press launches the downward attack. That would give it added utility allowing a target swap mid air for dodging bad etc. possibly even allow it to jump to allies ala shadowstep.
Perhaps too much value with all those changes though.