fix a stupid bug in my slight change to Prediction