fix issue in category graph generation
Merge pull request #16 from gsegura/master

fix issue while generating the category graph