fixed issue with doc on generator func