fix(test): Follow up test fix to target correct extension method.

fix(test): Follow up test fix to target correct extension method. #241