Fix the test failing