Fixing the compile errors of integration test.