Dirichlet distribution in 4.0.0b4

Quite surprising error. The Dirichlet has been refactored a long time ago and its used in several tests. I haven’t seen any issues with it.