fix bugs in augment data