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

@@ -98,11 +98,11 @@
                        InputWithCreateTimestamp
                      else
                        InputWithUpdateTimestamp
                      end
 
-            input(plugin.new(relation, input))
-            super(relation, options.merge(input: input))
+            wrapped_input = plugin.new(relation, options.fetch(:input) { input })
+            super(relation, options.merge(input: wrapped_input))
           end
         end
 
         # @since 0.7.0
         # @api private