fix for prediction round