(bug)Add generator bug fix