diff --git a/pyproject.toml b/pyproject.toml
index aecea33..41a840a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,7 +20,7 @@ documentation = "https://docs.graphbook.ai"
 repository = "https://github.com/graphbookai/graphbook"
 
 [tool.poetry.scripts]
-graphbook = "graphbook.server:main"
+graphbook = "graphbook.main:main"
 
 [tool.poetry.dependencies]
 python = "^3.10"