fixing up the tests