Fix bad codegen
Merge pull request #4 from dsyme/patch-3

Fix bad codegen