tweak for the new test framework