BUG: Predict didn't return the prediction