small fix in generator