lib/transflow/step_dsl.rb in transflow-0.0.2 vs lib/transflow/step_dsl.rb in transflow-0.1.0

@@ -12,10 +12,10 @@
 
     attr_reader :publish
 
     def initialize(name, options, container, steps, &block)
       @name = name
-      @handler = options.fetch(:with)
+      @handler = options.fetch(:with, name)
       @publish = options.fetch(:publish, false)
       @container = container
       @steps = steps
       instance_exec(&block) if block
     end