diff --git a/setup_empirical.py b/setup_empirical.py
index 5fdccb57..65e9d608 100644
--- a/setup_empirical.py
+++ b/setup_empirical.py
@@ -33,9 +33,7 @@
     long_description_content_type='text/plain',
     url='https://github.com/tensorflow/privacy',
     license='Apache-2.0',
-    packages=setuptools.find_packages(
-        where='privacy', include=['privacy_tests*']
-    ),
+    packages=setuptools.find_packages(include=['*privacy.privacy_tests*']),
     install_requires=[
         'absl-py>=1.0,==1.*',
         'immutabledict~=2.2',