From a391684159b5f967f159657b0237ef5100720f51 Mon Sep 17 00:00:00 2001
From: Alan Wu <XrXr@users.noreply.github.com>
Date: Thu, 4 Feb 2021 13:00:29 -0500
Subject: [PATCH] Check for iseq wrapper instead of iseq imemo

---
 ujit_iface.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ujit_iface.c b/ujit_iface.c
index 98698d4b79..bf36985ff6 100644
--- a/ujit_iface.c
+++ b/ujit_iface.c
@@ -353,7 +353,7 @@ iseqw_ujit_collect_blocks(st_data_t key, st_data_t value, st_data_t argp)
 static VALUE
 ujit_blocks_for(VALUE mod, VALUE rb_iseq)
 {
-    if (!rb_obj_is_iseq(rb_iseq)) {
+    if (CLASS_OF(rb_iseq) != rb_cISeq) {
         return rb_ary_new();
     }
     const rb_iseq_t *iseq = rb_iseqw_to_iseq(rb_iseq);
@@ -370,7 +370,7 @@ ujit_blocks_for(VALUE mod, VALUE rb_iseq)
 static VALUE
 ujit_install_entry(VALUE mod, VALUE iseq)
 {
-    if (!rb_obj_is_iseq(iseq)) {
+    if (CLASS_OF(iseq) != rb_cISeq) {
 	rb_raise(rb_eTypeError, "not an InstructionSequence");
     }
     rb_ujit_compile_iseq(rb_iseqw_to_iseq(iseq));