diff --git a/pyproject.toml b/pyproject.toml
index f79857c3d..6a539c132 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -130,9 +130,8 @@ Homepage = "https://github.com/meta-llama/llama-stack"
 llama = "llama_stack.cli.llama:main"
 install-wheel-from-presigned = "llama_stack.cli.scripts.run:install_wheel_from_presigned"
 
-[tool.setuptools]
-packages = { find = {} }
-license-files = []
+[tool.setuptools.packages.find]
+include = ["llama_stack"]
 
 [[tool.uv.index]]
 name = "pytorch-cpu"