fix integration test failure