remove tests that check for stage persistence in task since stage is no longer persisted in task