fixed up the tests a little bit