diff --git a/pyproject.toml b/pyproject.toml
index 03c1812..179e6e1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,8 +12,12 @@ dependencies = [
     "pydantic>=2.11.4",
     "aiohttp>=3.11.18",
 ]
-[tool.setuptools.packages.find]
-include = ["*_mcp_module"]
 
 [project.scripts]
-yargi-mcp = "yargi_mcp:__main__.main"
\ No newline at end of file
+yargi-mcp = "mcp_server_main:main"
+
+[tool.setuptools]
+py-modules = ["mcp_server_main"]
+
+[tool.setuptools.packages.find]
+include = ["*_mcp_module"]
\ No newline at end of file