diff --git a/py/runtime.c b/py/runtime.c
index 65d0df639e297928ebd4a044ec5e8add96b40e5a..54ec0d70b4f5bce6b63e45ceec7bd6a599a9bb8e 100644
--- a/py/runtime.c
+++ b/py/runtime.c
@@ -1215,13 +1215,12 @@ mp_vm_return_kind_t mp_resume(mp_obj_t self_in, mp_obj_t send_value, mp_obj_t th
 
     if (type->iternext != NULL && send_value == mp_const_none) {
         mp_obj_t ret = type->iternext(self_in);
+        *ret_val = ret;
         if (ret != MP_OBJ_STOP_ITERATION) {
-            *ret_val = ret;
             return MP_VM_RETURN_YIELD;
         } else {
             // Emulate raise StopIteration()
             // Special case, handled in vm.c
-            *ret_val = MP_OBJ_NULL;
             return MP_VM_RETURN_NORMAL;
         }
     }
diff --git a/py/vm.c b/py/vm.c
index 416de6b1a3283c0461f4064a22793f7e25eb3c31..2a8e3c990c7d99da68ce05a46aebd2acf04a0c5d 100644
--- a/py/vm.c
+++ b/py/vm.c
@@ -1178,8 +1178,7 @@ yield:
                     } else if (ret_kind == MP_VM_RETURN_NORMAL) {
                         // Pop exhausted gen
                         sp--;
-                        // TODO: When ret_value can be MP_OBJ_NULL here??
-                        if (ret_value == MP_OBJ_NULL || ret_value == MP_OBJ_STOP_ITERATION) {
+                        if (ret_value == MP_OBJ_STOP_ITERATION) {
                             // Optimize StopIteration
                             // TODO: get StopIteration's value
                             PUSH(mp_const_none);