Minor generator fix