Fixed the integration testcase.