diff --git a/mempalace/__init__.py b/mempalace/__init__.py
index f944507..212ac33 100644
--- a/mempalace/__init__.py
+++ b/mempalace/__init__.py
@@ -2,7 +2,6 @@
 
 import logging
 
-from .cli import main  # noqa: E402
 from .version import __version__  # noqa: E402
 
 # ChromaDB 0.6.x ships a Posthog telemetry client whose capture() signature is
@@ -25,4 +24,4 @@ logging.getLogger("chromadb.telemetry.product.posthog").setLevel(logging.CRITICA
 # intact, so the real fix is upgrading chromadb to 1.5.4+, which #581
 # proposes.  See #397 for the history of this line.
 
-__all__ = ["main", "__version__"]
+__all__ = ["__version__"]
diff --git a/pyproject.toml b/pyproject.toml
index 4ab1a1d..8700fd1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -35,7 +35,7 @@ Repository = "https://github.com/milla-jovovich/mempalace"
 "Bug Tracker" = "https://github.com/milla-jovovich/mempalace/issues"
 
 [project.scripts]
-mempalace = "mempalace:main"
+mempalace = "mempalace.cli:main"
 
 [project.optional-dependencies]
 dev = ["pytest>=7.0", "pytest-cov>=4.0", "ruff>=0.4.0", "psutil>=5.9"]