diff --git a/pyproject.toml b/pyproject.toml
index da99846..a62c58d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,6 +12,9 @@ requires-python = ">=3.10"
 readme = "README.md"
 license = {text = "MIT"}
 
+[project.scripts]
+litedoc = "litedoc.__main__:main"
+
 [build-system]
 requires = ["pdm-backend"]
 build-backend = "pdm.backend"
diff --git a/test_mkdoc.bat b/test_mkdoc.bat
index ab8e865..2181667 100644
--- a/test_mkdoc.bat
+++ b/test_mkdoc.bat
@@ -1,3 +1,3 @@
 chcp 65001
-python -m litedoc tests/test_modules/mbcp -o docdist/api -l zh-Hans -fd def -md def -cd 类 -cs
-python -m litedoc tests/test_modules/mbcp -o docdist/en/api -l en -fd def -md def -cs
\ No newline at end of file
+litedoc tests/test_modules/mbcp -o docdist/api -l zh-Hans -fd def -md def -cd 类 -cs
+litedoc tests/test_modules/mbcp -o docdist/en/api -l en -fd def -md def -cs
\ No newline at end of file