Fix integration test after merging pull request