lib/hanami/model/plugins/mapping.rb in hanami-model-1.0.0 vs lib/hanami/model/plugins/mapping.rb in hanami-model-1.0.1

@@ -35,11 +35,11 @@
           # Builds the output processor
           #
           # @since 0.7.0
           # @api private
           def build(relation, options = {})
-            input(InputWithMapping.new(relation, input))
-            super(relation, options.merge(input: input))
+            wrapped_input = InputWithMapping.new(relation, options.fetch(:input) { input })
+            super(relation, options.merge(input: wrapped_input))
           end
         end
 
         # @since 0.7.0
         # @api private