correction to cifar_prediction