fix up the tests
fixup! fix up the tests