update the tests a little