diff --git a/setup.py b/setup.py
index 94d96cd13..91ea3d8a1 100644
--- a/setup.py
+++ b/setup.py
@@ -195,6 +195,7 @@ def build_extensions(self):
         long_description_content_type="text/markdown",
         url="https://github.com/Chia-Network/chiapos",
         setup_requires=["pybind11"],
+        tests_require=["pytest"],
         ext_modules=ext_modules,
         cmdclass={"build_ext": BuildExt},
         zip_safe=False,
@@ -210,6 +211,7 @@ def build_extensions(self):
         long_description=open("README.md").read(),
         long_description_content_type="text/markdown",
         url="https://github.com/Chia-Network/chiapos",
+        tests_require=["pytest"],
         ext_modules=[CMakeExtension("chiapos", ".")],
         cmdclass=dict(build_ext=CMakeBuild),
         zip_safe=False,