Remove null references to stage for munit tests.