diff --git a/MANIFEST.in b/MANIFEST.in
deleted file mode 100644
index 0617953..0000000
--- a/MANIFEST.in
+++ /dev/null
@@ -1 +0,0 @@
-include kfac_jax/py.typed
diff --git a/pyproject.toml b/pyproject.toml
index 0cb1936..a512503 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -74,3 +74,6 @@ tests = [  # these should be version pinned?
   "dm-tree>=0.1.7",
   "optax>=0.1.4",
 ]
+
+[tool.setuptools.package-data]
+"*" = ["py.typed"]