testPrediction fix