diff --git a/setup.py b/setup.py
index 066841a..39f48e6 100755
--- a/setup.py
+++ b/setup.py
@@ -16,7 +16,7 @@
 # desiutil needs to import some of its own code.
 #
 sys.path.insert(int(sys.path[0] == ''),os.path.abspath('./py'))
-from desiutil.setup import DesiModule, DesiTest, DesiVersion, get_version
+from desiutil.setup import DesiModule, DesiVersion, get_version
 #
 # Begin setup
 #
@@ -57,7 +57,7 @@
 setup_keywords['use_2to3'] = False
 setup_keywords['packages'] = find_packages('py')
 setup_keywords['package_dir'] = {'':'py'}
-setup_keywords['cmdclass'] = {'module_file': DesiModule, 'version': DesiVersion, 'test': DesiTest, 'sdist': DistutilsSdist}
+setup_keywords['cmdclass'] = {'module_file': DesiModule, 'version': DesiVersion, 'sdist': DistutilsSdist}
 setup_keywords['test_suite']='{name}.test.test_suite'.format(**setup_keywords)
 #
 # Autogenerate command-line scripts.