Fix a couple errors in the generator