Changed group augment to use new api