bug in predict fixed