Address losing sub_arg during early return.
Address losing sub_arg during early return.

Instead of popping immediately, only pop after the step function
launches w/o an error.

jhuapl-boss/heaviside,jhuapl-boss/heaviside