Adjust the tests