Small change in generator