If you want the value only at the point of your data you could pass your original X into the conditional instead of X_new