Make the generator work when installed from NPM