The inability to keep upgrades at each try makes the "story mode" look more like an "arcade mode". Either have some form of "power retention" or call it what it is: an arcade mode.
I mean, a continue credit system? That has Arcade written all over it.