diff --git a/python/setup-css.py b/python/setup-css.py
index 1b8af16cc..e35341b3a 100755
--- a/python/setup-css.py
+++ b/python/setup-css.py
@@ -6,27 +6,6 @@
 from setuptools import setup
 from cssbeautifier.__version__ import __version__
 
-from setuptools.command.test import test as TestCommand
-
-DIR_CSS = "cssbeautifier/tests/"
-
-
-class PyTestCSS(TestCommand):
-    user_options = [("pytest-args=", "a", "Arguments to pass to py.test")]
-
-    def initialize_options(self):
-        TestCommand.initialize_options(self)
-        self.pytest_args = ["--assert=plain"] + [
-            DIR + x for x in os.listdir(DIR) if x.endswith(".py") and x[0] not in "._"
-        ]
-
-    def run_tests(self):
-        # import here, cause outside the eggs aren't loaded
-        import pytest
-
-        errno = pytest.main(self.pytest_args)
-        sys.exit(errno)
-
 
 setup(
     name="cssbeautifier",
@@ -46,6 +25,4 @@ def run_tests(self):
     ],
     install_requires=["jsbeautifier", "six>=1.13.0", "editorconfig>=0.12.2"],
     license="MIT",
-    test_suite="pytest.collector",
-    cmdclass={"test": PyTestCSS},
 )
diff --git a/python/setup-js.py b/python/setup-js.py
index 75f29ab78..2d716bcfd 100755
--- a/python/setup-js.py
+++ b/python/setup-js.py
@@ -3,6 +3,7 @@
 import os
 import sys
 
+
 from setuptools import setup
 from jsbeautifier.__version__ import __version__
 
@@ -49,6 +50,4 @@ def run_tests(self):
     ],
     install_requires=["six>=1.13.0", "editorconfig>=0.12.2"],
     license="MIT",
-    test_suite="pytest.collector",
-    cmdclass={"test": PyTest},
 )