fix predict