This seems to be working properly for me with several distributions and dimensions. But I think other distributions and multi-dimensional cases should be tested.