diff --git a/MANIFEST.in b/MANIFEST.in
index 4cc82c5dd..365d65ff3 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -4,3 +4,5 @@ include requirements.txt
 include requirements-jax.txt
 include requirements-test.txt
 include haiku/py.typed
+
+global-exclude *_test.py
diff --git a/setup.py b/setup.py
index 25b39b183..7fd8c5ad9 100644
--- a/setup.py
+++ b/setup.py
@@ -46,7 +46,7 @@ def _parse_requirements(requirements_txt_path):
     long_description_content_type='text/markdown',
     author_email='haiku-dev-os@google.com',
     # Contained modules and scripts.
-    packages=find_namespace_packages(exclude=['*_test.py', 'examples']),
+    packages=find_namespace_packages(exclude=['examples*', 'docs*']),
     install_requires=_parse_requirements('requirements.txt'),
     extras_require={'jax': _parse_requirements('requirements-jax.txt')},
     tests_require=_parse_requirements('requirements-test.txt'),