From 4ab801c045b2815b6191188c2eec03816203604d Mon Sep 17 00:00:00 2001 From: mirichi Date: Wed, 23 Apr 2014 19:39:14 +0900 Subject: fixed. --- src/vm.c | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src') diff --git a/src/vm.c b/src/vm.c index 636cc5ac5..77910b1bf 100644 --- a/src/vm.c +++ b/src/vm.c @@ -382,7 +382,7 @@ mrb_funcall_with_block(mrb_state *mrb, mrb_value self, mrb_sym mid, int argc, co } else { ci->nregs = p->body.irep->nregs + n; - stack_extend(mrb, ci->nregs, p->body.irep->nlocals, 0); + stack_extend(mrb, ci->nregs, p->body.irep->nlocals, argc+2); } mrb->c->stack[0] = self; if (undef) { @@ -513,7 +513,7 @@ mrb_yield_with_class(mrb_state *mrb, mrb_value b, int argc, const mrb_value *arg } else { ci->nregs = p->body.irep->nregs + 1; - stack_extend(mrb, ci->nregs, p->body.irep->nlocals, 0); + stack_extend(mrb, ci->nregs, p->body.irep->nlocals, argc+2); } mrb->c->stack[0] = self; -- cgit v1.2.3