lib/templates/default/steptransformers/setup.rb in cucumber-in-the-yard-1.6.3 vs lib/templates/default/steptransformers/setup.rb in cucumber-in-the-yard-1.6.4

@@ -6,22 +6,24 @@
 
 def stepdefinitions
   @item_title = "Step Definitions"
   @item_type = "step definition"
   @items = YARD::Registry.all(:stepdefinition)
-  erb(:transformers)
+  erb(:header) + erb(:transformers)
 end
 
 def steptransforms
   @item_title = "Step Transforms"
   @item_type = "step transform"
   @items = YARD::Registry.all(:steptransform)
-  erb(:transformers)
+  erb(:header) + erb(:transformers)
 end
 
 def undefined_steps
+  @item_title = "Undefined Steps"
+  @item_type = "undefined steps"
   @undefined_steps ||= Registry.all(:step).reject {|s| s.definition || s.scenario.outline? }
-  erb(:undefined_steps)
+  erb(:header) + erb(:undefined_steps)
 end
 
 
 def link_transformed_step(step)
   value = step.value
\ No newline at end of file