minor bug
fixed bug in augment
0.7.3