Tiny context cleanup in the tests.