fix the tests a bit