Adapt integration tests