Fix weight issue
Merge pull request #19 from gpetruc/fix-non-uniform-binned-generation

Fix weight issue