Fixed augmenter issue