Adjust the test
Adjust the tests