Fix the failure of the integration test.

Fix the failure of the integration test.