Partially upgrades integration tests