fix up the tests yo