diff --git a/pyproject.toml b/pyproject.toml
index baa33c6..03c1812 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,4 +16,4 @@ dependencies = [
 include = ["*_mcp_module"]
 
 [project.scripts]
-yargi-mcp = "mcp_server_main:main"
\ No newline at end of file
+yargi-mcp = "yargi_mcp:__main__.main"
\ No newline at end of file