Updated test to reflect new version