Update the addition test to match the new style