Fix for integration test in CI