fix flaky integration test