Fix a bug in the predictor and added more tests