Minor tweak for tests.