Fixing Prediction

Small change to make it work :P