Fixing a small code-gen bug