Fix compile errors in unit test.