fixed up the tests