Minor syntactic touch ups in the augment module