1 files changed, 6 insertions, 1 deletions
diff --git a/lib/ruby_vm/mjit/compiler.rb b/lib/ruby_vm/mjit/compiler.rb
index 79a5da44a2..30f2557f5b 100644
--- a/lib/ruby_vm/mjit/compiler.rb
+++ b/lib/ruby_vm/mjit/compiler.rb
@@ -133,6 +133,12 @@ module RubyVM::MJIT
       list_blocks(iseq, pc).each do |block|
         invalidate_block(iseq, block)
       end
+
+      # If they were the ISEQ's first blocks, re-compile MJIT entry as well
+      if iseq.body.iseq_encoded.to_i == pc
+        iseq.body.jit_func = 0
+        iseq.body.total_calls = 0
+      end
     end
 
     private
@@ -255,7 +261,6 @@ module RubyVM::MJIT
           @cb.write(branch_asm)
         end
       end
-      # TODO: Reset jit_func and total_calls if it's the first block after prelude
     end
 
     def list_blocks(iseq, pc)