Fix the bug in the test.