diff --git a/setup.py b/setup.py
index 9bfe948..2e9c5d8 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name="jaxlie",
-    version="0.0.3",
+    version="0.0.4",
     description="Matrix Lie groups in Jax",
     long_description=long_description,
     long_description_content_type="text/markdown",
@@ -14,7 +14,7 @@
     author_email="brentyi@berkeley.edu",
     license="MIT",
     packages=find_packages(),
-    package_data={"liejax": ["py.typed"]},
+    package_data={"jaxlie": ["py.typed"]},
     python_requires=">=3.6",
     install_requires=[
         "flax",