diff --git a/pyproject.toml b/pyproject.toml
index 7411e4e7..2c63459b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -24,3 +24,6 @@ isort = "^4.3"
 
 [tool.black]
 line-length = 120
+
+[tool.poetry.scripts]
+diagrams = 'scripts.generate:main'
diff --git a/scripts/generate.py b/scripts/generate.py
index 4e3ef425..785909c5 100644
--- a/scripts/generate.py
+++ b/scripts/generate.py
@@ -94,9 +94,11 @@ def generate(pvd: str) -> None:
     apidoc = gen_apidoc(pvd, typ_paths)
     make_apidoc(pvd, apidoc)
 
-
-if __name__ == "__main__":
+def main():
     pvd = sys.argv[1]
     if pvd not in cfg.PROVIDERS:
         sys.exit()
     generate(pvd)
+
+if __name__ == "__main__":
+    main()