Fix predictor NaN error