diff --git a/pyproject.toml b/pyproject.toml
index a276f11..cdd11e1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -26,6 +26,7 @@ dynamic = ["version"]
 
 [tool.setuptools]
 packages = ["dataretrieval", "dataretrieval.codes"]
+package-data.dataretrieval = ["py.typed"]
 
 [project.optional-dependencies]
 test = [
diff --git a/setup.py b/setup.py
index 3630daf..6068493 100644
--- a/setup.py
+++ b/setup.py
@@ -1,5 +1,3 @@
 from setuptools import setup
 
-setup(
-    package_data={"dataretrieval": ["py.typed"]}
-)
\ No newline at end of file
+setup()