Fix to DistributionMatrix due to deprecation warning