Fixed the test failure.