Added upgrading tests