Fix compile error in unit test.