diff --git a/pyproject.toml b/pyproject.toml
index 4e398c60a..ab04fb5f8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -132,7 +132,8 @@ llama = "llama_stack.cli.llama:main"
 install-wheel-from-presigned = "llama_stack.cli.scripts.run:install_wheel_from_presigned"
 
 [tool.setuptools.packages.find]
-include = ["llama_stack"]
+where = ["."]
+include = ["llama_stack", "llama_stack.*"]
 
 [[tool.uv.index]]
 name = "pytorch-cpu"