started fixing ClassGeneratorTest