small bug fix, pred array size