From b4b7a0f2f9e1e439d09a4189f3d7bf0155141490 Mon Sep 17 00:00:00 2001
From: Hao Zhang <hzhangxyz@outlook.com>
Date: Sun, 15 Mar 2026 17:41:47 +0800
Subject: [PATCH] chore: Update generator.hh for better code quality.

---
 apyds/ds.cc | 2 +-
 atsds/ds.cc | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/apyds/ds.cc b/apyds/ds.cc
index 142914f..b1b80bf 100644
--- a/apyds/ds.cc
+++ b/apyds/ds.cc
@@ -18,7 +18,7 @@ class Iterator {
             iterator = std::make_unique<iterator_t>(generator.begin());
             initialized = true;
         }
-        if (*iterator == nullptr) {
+        if (*iterator == generator.end()) {
             return nullptr;
         }
         ds::rule_t* result = **iterator;
diff --git a/atsds/ds.cc b/atsds/ds.cc
index 03bbe29..f94f84c 100644
--- a/atsds/ds.cc
+++ b/atsds/ds.cc
@@ -17,7 +17,7 @@ class Iterator {
             iterator = std::make_unique<iterator_t>(generator.begin());
             initialized = true;
         }
-        if (*iterator == nullptr) {
+        if (*iterator == generator.end()) {
             return nullptr;
         }
         ds::rule_t* result = **iterator;