need to debug validation with augmentation.