Remove use of deprecated addAugmentation()