diff --git a/.travis.yml b/.travis.yml
index 5139ad8..2c09e50 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -21,4 +21,4 @@ install:
   - "pip install $SQLALCHEMY"
   - pip install -e ".[test]"
 script:
-  - python setup.py test
+  - py.test
diff --git a/setup.py b/setup.py
index a49146f..c457e5e 100644
--- a/setup.py
+++ b/setup.py
@@ -4,25 +4,10 @@ SQLAlchemy-Utils
 
 Various utility functions and custom data types for SQLAlchemy.
 """
-from setuptools import setup, Command, find_packages
-import subprocess
+from setuptools import setup, find_packages
 import sys
 
 
-class PyTest(Command):
-    user_options = []
-
-    def initialize_options(self):
-        pass
-
-    def finalize_options(self):
-        pass
-
-    def run(self):
-        errno = subprocess.call(['py.test'])
-        sys.exit(errno)
-
-
 PY3 = sys.version_info[0] == 3
 
 
@@ -85,7 +70,6 @@ setup(
         if sys.version_info[0] == 2 and sys.version_info[1] < 7 else ''
     ],
     extras_require=extras_require,
-    cmdclass={'test': PyTest},
     classifiers=[
         'Environment :: Web Environment',
         'Intended Audience :: Developers',