Fixed get_distribution