fix the tests up