Fixed saving the distribution