GeneratorTest fix