diff --git a/setup.py b/setup.py
index d534cec4..801d8d50 100644
--- a/setup.py
+++ b/setup.py
@@ -404,7 +404,7 @@ setup(
         "Topic :: Scientific/Engineering :: Artificial Intelligence",
     ],
     packages=find_packages(exclude=("benchmarks", "csrc", "docs", "examples",
-                                    "tests")),
+                                    "tests*")),
     python_requires=">=3.8",
     install_requires=get_requirements(),
     ext_modules=ext_modules,