Fix the warning in test.