lib/draper/test_case.rb in draper-3.0.1 vs lib/draper/test_case.rb in draper-3.1.0

@@ -1,12 +1,12 @@
 module Draper
   require 'active_support/test_case'
 
   class TestCase < ::ActiveSupport::TestCase
     module ViewContextTeardown
-      def teardown
-        super
+      def before_setup
         Draper::ViewContext.clear!
+        super
       end
     end
 
     module Behavior
       if defined?(::Devise)