Prediction is going wrong