unit test upgrades