Minor generator bugfix.