Fix tests

Use updated not upgrade.