Fixed the failed test.