Fixed the test failure