API change on Predictor