refactor

1 files changed, 5 insertions, 5 deletions

diff --git a/src/distutils2/command/test.py b/src/distutils2/command/test.py
index 822e459..bdaf1d6 100644
--- a/src/distutils2/command/test.py
+++ b/src/distutils2/command/test.py
@@ -1,7 +1,7 @@
 import os, sys
 from distutils2.core import Command 
 from distutils2._backport.pkgutil import get_distribution
-from distutils2.util import resolve_dotted_name
+from distutils2.util import resolve_name
 import unittest
 import warnings
 
@@ -36,11 +36,11 @@ class test(Command):
                 self.run_command('build')
                 os.chdir(self.build_lib)
             args = {"module": self.test_suite,
-                    "argv": sys.argv[:1],
-                    "testLoader": resolve_dotted_name(self.test_loader)
+                    "argv": sys.argv[:1]
             }
-            if args['testLoader'] is None:
-                del args['testLoader']
+            loader_instance = resolve_name(self.test_loader)
+            if loader_instance is not None:
+                args['testLoader'] = loader_instance
             unittest.main(**args)
         finally:
             os.chdir(prev_cwd)