diff --git a/test/functional/application_controller_test.rb b/test/functional/application_controller_test.rb
index 47e0e41..6c6448f 100644
--- a/test/functional/application_controller_test.rb
+++ b/test/functional/application_controller_test.rb
@@ -160,16 +160,13 @@ class ApplicationControllerTest < ActionController::TestCase
     assert !DoesNotUsesBlocksTestController.new.uses_design_blocks?
   end
 
-  should 'use design plugin to generate blocks' do
+  should 'generate blocks' do
     get :index
     assert_tag :tag => 'div', :attributes => { :id => 'boxes', :class => 'boxes' }
   end
 
-  should 'not use design plugin when tells so' do
-    class NoDesignBlocksTestController < ApplicationController
-      no_design_blocks
-    end
-    @controller = NoDesignBlocksTestController.new
+  should 'not generate blocks when told not to do so' do
+    @controller.stubs(:uses_design_blocks?).returns(false)
     get :index
     assert_no_tag :tag => 'div', :attributes => { :id => 'boxes', :class => 'boxes'  }
   end
--
libgit2 0.21.2