2 files changed, 10 insertions, 10 deletions
diff --git a/actionpack/test/template/body_parts_test.rb b/actionpack/test/template/body_parts_test.rb
index 4c82b75cdc..5be8533293 100644
--- a/actionpack/test/template/body_parts_test.rb
+++ b/actionpack/test/template/body_parts_test.rb
@@ -6,7 +6,7 @@ class BodyPartsTest < ActionController::TestCase
   class TestController < ActionController::Base
     def index
       RENDERINGS.each do |rendering|
-        response.template.punctuate_body! rendering
+        @template.punctuate_body! rendering
       end
       @performed_render = true
     end
diff --git a/actionpack/test/template/output_buffer_test.rb b/actionpack/test/template/output_buffer_test.rb
index 6d8eab63dc..bc17f36783 100644
--- a/actionpack/test/template/output_buffer_test.rb
+++ b/actionpack/test/template/output_buffer_test.rb
@@ -13,23 +13,23 @@ class OutputBufferTest < ActionController::TestCase
     # Start with the default body parts
     get :index
     assert_equal ['foo'], @response.body_parts
-    assert_nil @response.template.output_buffer
+    assert_nil @controller.template.output_buffer
 
     # Nil output buffer is skipped
-    @response.template.flush_output_buffer
-    assert_nil @response.template.output_buffer
+    @controller.template.flush_output_buffer
+    assert_nil @controller.template.output_buffer
     assert_equal ['foo'], @response.body_parts
 
     # Empty output buffer is skipped
-    @response.template.output_buffer = ''
-    @response.template.flush_output_buffer
-    assert_equal '', @response.template.output_buffer
+    @controller.template.output_buffer = ''
+    @controller.template.flush_output_buffer
+    assert_equal '', @controller.template.output_buffer
     assert_equal ['foo'], @response.body_parts
 
     # Flushing appends the output buffer to the body parts
-    @response.template.output_buffer = 'bar'
-    @response.template.flush_output_buffer
-    assert_equal '', @response.template.output_buffer
+    @controller.template.output_buffer = 'bar'
+    @controller.template.flush_output_buffer
+    assert_equal '', @controller.template.output_buffer
     assert_equal ['foo', 'bar'], @response.body_parts
   end
 end