Minor updates to augments.