diff --git a/setup.py b/setup.py
index 456ecc61..fa90b095 100644
--- a/setup.py
+++ b/setup.py
@@ -117,8 +117,7 @@ def _make_docs_packages():
     python_requires='>=3.9,<4',
     packages=find_packages(),
     include_package_data=True,
-    package_data={'tensorflow_transform': ['py.typed']},
-    data_files=[("docs_reqs", ["requirements-docs.txt"])],
+    package_data={'tensorflow_transform': ['py.typed', 'requirements-docs.txt']},
     description='A library for data preprocessing with TensorFlow',
     long_description=_LONG_DESCRIPTION,
     long_description_content_type='text/markdown',