Fix the test failure