lib/arstotzka/wrapper.rb in arstotzka-1.3.0 vs lib/arstotzka/wrapper.rb in arstotzka-1.3.1

@@ -64,11 +64,11 @@
 
     private
 
     # @private
     attr_reader :options
-    delegate :klass, :type, to: :options
+    delegate :klass, :type, :after_each, to: :options
 
     # @private
     #
     # Wraps an element with a class and perform typecasting
     #
@@ -88,11 +88,11 @@
     #
     # @see #wrap_element
     #
     # @return [Arra]
     def wrap_array(array)
-      array.map { |v| wrap v }
+      array.map { |element| wrap element }
     end
 
     # @private
     #
     # Check if type was given
@@ -130,11 +130,11 @@
     #
     # Process and wrap value trhough a method call
     #
     # @return [Object] result of method call
     def after(value)
-      return value unless options.after_each
+      return value unless after_each
 
-      options.instance.send(options.after_each, value)
+      options.instance.send(after_each, value)
     end
   end
 end