predict bug fixed