Minor tweak for tests