Fixed bug in Dirichlet example