generator bug fix?