Fix a bug in predict() method