fixed predict in util.py