Fix errot on dictribution