Fix the unit test.
Fix the unit test.