fix predict_args