Adjust stack position of NODE_RESCUE.

1 files changed, 4 insertions, 0 deletions

diff --git a/src/codegen.c b/src/codegen.c
index 86d61fecf..d3239259b 100644
--- a/src/codegen.c
+++ b/src/codegen.c
@@ -1083,6 +1083,7 @@ codegen(codegen_scope *s, node *tree, int val)
           }
           if (n3->cdr->cdr->car) {
             codegen(s, n3->cdr->cdr->car, val);
+            if (val) pop();
           }
           tmp = new_label(s);
           genop(s, MKOP_sBx(OP_JMP, exend));
@@ -1102,6 +1103,9 @@ codegen(codegen_scope *s, node *tree, int val)
       if (tree->car) {
         codegen(s, tree->car, val);
       }
+      else if (val) {
+        push();
+      }
       dispatch_linked(s, exend);
       loop_pop(s, NOVAL);
     }