Fixed a small bug on the generator

...eeded only.