Fixed compilation errors in unit test.