fix integration test.