From 0f7dce130cd19eb8b6175b647f6b2f27c90fcb3a Mon Sep 17 00:00:00 2001
From: blueloveTH <blueloveTH@foxmail.com>
Date: Wed, 19 Apr 2023 13:45:09 +0800
Subject: [PATCH] ...

---
 src/iter.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/iter.h b/src/iter.h
index b392c4aa..94c59548 100644
--- a/src/iter.h
+++ b/src/iter.h
@@ -82,6 +82,7 @@ inline PyObject* Generator::next(){
         for(PyObject* obj: frame.stack_view()) s_backup.push_back(obj);
         vm->_pop_frame();
         state = 1;
+        if(ret == vm->StopIteration) state = 2;
         return ret;
     }else{
         state = 2;