diff --git a/pyproject.toml b/pyproject.toml
index ef0b50f..c09df5f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -2,10 +2,13 @@
 name = "comfyui-to-python-extension"
 description = "This custom node allows you to generate pure python code from your ComfyUI workflow with the click of a button. Great for rapid experimentation or production deployment."
 version = "2.1.0"
-license = { text = "MIT License" }
+license = "MIT"
 requires-python = ">=3.12"
 dependencies = ["black"]
 
+[tool.setuptools.packages.find]
+include = ["comfyui_to_python*"]
+
 [project.optional-dependencies]
 dev = ["ruff"]