1 files changed, 6 insertions, 3 deletions
diff --git a/array.rb b/array.rb
index 5f31693cab..81beff0b1c 100644
--- a/array.rb
+++ b/array.rb
@@ -212,7 +212,7 @@ class Array
     indexes
   end
 
-  with_yjit do
+  with_jit do
     if Primitive.rb_builtin_basic_definition_p(:each)
       undef :each
 
@@ -245,7 +245,8 @@ class Array
         value = nil
         result = Primitive.ary_sized_alloc
         while Primitive.cexpr!(%q{ ary_fetch_next(self, LOCAL_PTR(_i), LOCAL_PTR(value)) })
-          result << yield(value)
+          value = yield(value)
+          Primitive.cexpr!(%q{ rb_ary_push(result, value) })
         end
         result
       end
@@ -270,7 +271,9 @@ class Array
         value = nil
         result = Primitive.ary_sized_alloc
         while Primitive.cexpr!(%q{ ary_fetch_next(self, LOCAL_PTR(_i), LOCAL_PTR(value)) })
-          result << value if yield value
+          if yield value
+            Primitive.cexpr!(%q{ rb_ary_push(result, value) })
+          end
         end
         result
       end