diff --git a/test/unit/cors_test.rb b/test/unit/cors_test.rb
index 5360613..00d6431 100644
--- a/test/unit/cors_test.rb
+++ b/test/unit/cors_test.rb
@@ -4,7 +4,7 @@
 require 'rack/lint'
 require 'minitest/autorun'
 require 'rack/test'
-require 'mocha/setup'
+require 'mocha/minitest'
 require 'rack/cors'
 require 'ostruct'
 
@@ -21,7 +21,7 @@ def options(uri, params = {}, env = {}, &block)
   def_delegator :current_session, :options
 end
 
-module MiniTest::Assertions
+module Minitest::Assertions
   def assert_cors_success(response)
     assert !response.headers['Access-Control-Allow-Origin'].nil?, 'Expected a successful CORS response'
   end