Fixed writing info in predict mode