test/test_helper.rb in haml-rails-1.0.0 vs test/test_helper.rb in haml-rails-2.0.0

@@ -71,9 +71,9 @@
 alias :require_generator :require_generators
 
 require_generators generator_list
 
 # Remove tmp directory when test suite is completed
-MiniTest::Unit.after_tests do
+MiniTest.after_run do
   tmp_dir = File.expand_path(File.join(File.dirname(__FILE__), '..', 'tmp'))
   FileUtils.rm_r(tmp_dir)
 end