Correct the tests