Correct minor issue with dirichlet sample size