Prediction bug fixed