Bugfix to sample_dirichlet_safe