Fix new generator issue.