diff --git a/Python/optimizer_analysis.c b/Python/optimizer_analysis.c
index d635ebabf9007a..fd668206cb1a24 100644
--- a/Python/optimizer_analysis.c
+++ b/Python/optimizer_analysis.c
@@ -468,7 +468,7 @@ optimize_uops(
             ctx->out_buffer[ctx->out_len++] = *this_instr;
         }
         assert(ctx->frame != NULL);
-        if (!CURRENT_FRAME_IS_INIT_SHIM()) {
+        if (!CURRENT_FRAME_IS_INIT_SHIM() && !ctx->done) {
             DPRINTF(3, " stack_level %d\n", STACK_LEVEL());
             ctx->frame->stack_pointer = stack_pointer;
             assert(STACK_LEVEL() >= 0);