Fix the bug of prediction initialization