fix for prediction