Update integration test after merge