tau_prediction removed