diff --git a/pyproject.toml b/pyproject.toml
index 35be1de..b7a280f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -48,5 +48,6 @@ comfycli = "comfy_cli.__main__:main"
 [project.urls]
 Repository = "https://github.com/drip-art/comfy-cli.git"
 
-
+[tool.setuptools]
+py-modules = ["comfy_cli"]