prediction fix pt. 2