Dirichlet distribution in 4.0.0b4

Hi @ricardoV94! I was surprised by this problem as well and am still not entirely sure there is nothing wrong on my end. Can you reproduce the error?