fix an issue with the code generation
Merge pull request #206 from devoncarew/devoncarew_fix_generation

fix an issue with the code generation