fixed bug in pred.py