Modify the tests