fix test to match with other updates