Fix the test failing.