Upgrade test
Fix tests

Use updated not upgrade.