minor update for endoPredict