fixed issue in generator