diff --git a/test/unit/disabled_enterprise_message_block_test.rb b/test/unit/disabled_enterprise_message_block_test.rb
index 16ddb4a..c6997a2 100644
--- a/test/unit/disabled_enterprise_message_block_test.rb
+++ b/test/unit/disabled_enterprise_message_block_test.rb
@@ -19,12 +19,17 @@ class DisabledEnterpriseMessageBlockTest < ActiveSupport::TestCase
     block = enterprise.boxes.first.blocks.first
 
     ApplicationHelper.class_eval do
+      alias_method :original_profile, :profile
      def profile
        return Enterprise['disabled-enterprise']
      end
     end
 
     assert_match 'This message is for disabled enterprises', render_block_content(block)
+
+    ApplicationHelper.class_eval do
+      alias_method :profile, :original_profile
+    end
   end
 
   should 'not be editable' do
--
libgit2 0.21.2