Generator fixes.