Fix deprecation warnings around addAugmentation()