diff --git a/lib/temple/generators.rb b/lib/temple/generators.rb
index 1f5eb82..188ba8d 100644
--- a/lib/temple/generators.rb
+++ b/lib/temple/generators.rb
@@ -174,7 +174,8 @@ def on_dynamic(code)
     #   _buf
     class RailsOutputBuffer < StringBuffer
       set_default_options :buffer_class => 'ActionView::OutputBuffer',
-                          :buffer => '@output_buffer'
+                          :buffer => '@output_buffer',
+                          :capture_generator => RailsOutputBuffer
 
       def preamble
         "#{buffer} = #{options[:buffer_class]}.new"
@@ -183,10 +184,6 @@ def preamble
       def concat(str)
         "#{buffer}.safe_concat((#{str}))"
       end
-
-      def on_capture(name, block)
-        ["#{name} = #{buffer}", compile!(block), name].join(' ; ')
-      end
     end
   end