Updating test to match