move no block check to mrb_yield_internal

1 files changed, 3 insertions, 0 deletions

diff --git a/src/vm.c b/src/vm.c
index 13058fc28..9224c7a40 100644
--- a/src/vm.c
+++ b/src/vm.c
@@ -392,6 +392,9 @@ mrb_yield_internal(mrb_state *mrb, mrb_value b, int argc, mrb_value *argv, mrb_v
   int n = mrb->ci->nregs;
   mrb_value val;
 
+  if (mrb_nil_p(b)) {
+    mrb_raise(mrb, E_ARGUMENT_ERROR, "no block given");
+  }
   p = mrb_proc_ptr(b);
   ci = cipush(mrb);
   ci->mid = mid;