Make exit_after_finished actually return the wrapper