fixes to predictor