Fix integration testing