Fixed class for SimpleTextGeneration