Define `#+` to `Enumerator` and `Enumerator#Chain` instead of `Enumerable`

1 files changed, 7 insertions, 3 deletions

diff --git a/mrbgems/mruby-enum-chain/mrblib/chain.rb b/mrbgems/mruby-enum-chain/mrblib/chain.rb
index 52f5f0656..03acfb7ef 100644
--- a/mrbgems/mruby-enum-chain/mrblib/chain.rb
+++ b/mrbgems/mruby-enum-chain/mrblib/chain.rb
@@ -6,13 +6,13 @@ module Enumerable
   def chain(*args)
     Enumerator::Chain.new(self, *args)
   end
+end
 
+class Enumerator
   def +(other)
-    Enumerator::Chain.new(self, other)
+    Chain.new(self, other)
   end
-end
 
-class Enumerator
   class Chain
     include Enumerable
 
@@ -50,6 +50,10 @@ class Enumerator
       self
     end
 
+    def +(other)
+      self.class.new(self, other)
+    end
+
     def inspect
       "#<#{self.class}: #{@enums.inspect}>"
     end