.autotest in ZenTest-4.0.0 vs .autotest in ZenTest-4.1.0

@@ -7,9 +7,13 @@
   at.add_exception 'coverage'
   at.add_exception 'coverage.info'
 
   at.libs << ':../../minitest/dev/lib'
   at.testlib = "minitest/autorun"
+
+  %w(TestZenTest).each do |klass|
+    at.extra_class_map[klass] = "test/test_zentest.rb"
+  end
 end
 
 Autotest.add_hook :all_good do |at|
   system "rake rcov_info"
 end if ENV['RCOV']