diff --git a/mjx/pyproject.toml b/mjx/pyproject.toml
index fbf96bcf..08e01e0a 100644
--- a/mjx/pyproject.toml
+++ b/mjx/pyproject.toml
@@ -79,3 +79,6 @@ extend-exclude = '''(
 norecursedirs = [
     "**/third_party",
 ]
+
+[tool.setuptools.packages.find]
+include = ["mujoco*"]