prediction_calculator fix