Upgrade the generator. Fixes gh-1