Fix plotting of distributions