fix model to return epsilon instead of p