fix predict bug