1 files changed, 3 insertions, 3 deletions
diff --git a/Python/compile.c b/Python/compile.c
index ce714dce6e..9fc997cdf5 100644
--- a/Python/compile.c
+++ b/Python/compile.c
@@ -7162,9 +7162,6 @@ stackdepth(basicblock *entryblock, int code_flags)
                 next = NULL;
                 break;
             }
-            if (instr->i_opcode == YIELD_VALUE) {
-                instr->i_oparg = depth;
-            }
         }
         if (next != NULL) {
             assert(BB_HAS_FALLTHROUGH(b));
@@ -7332,6 +7329,9 @@ label_exception_targets(basicblock *entryblock) {
                 }
             }
             else {
+                if (instr->i_opcode == YIELD_VALUE) {
+                    instr->i_oparg = except_stack->depth;
+                }
                 instr->i_except = handler;
             }
         }