diff --git a/test/test_helper.rb b/test/test_helper.rb
index e3eebae4..eee2d049 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -17,9 +17,10 @@ Minitest::Test = MiniTest::Unit::TestCase unless defined?(Minitest::Test)
 require 'capture_warnings'
 @capture_warnings = CaptureWarnings.new(fail_build = false)
 @capture_warnings.before_tests
-at_exit do
+Minitest.after_run do
   @capture_warnings.after_tests
 end
+
 require 'active_model_serializers'
 
 require 'support/stream_capture'