Fixed the issue of test failure