Updated test expected data to match results after fix