diff --git a/cli.py b/demo/demo_cli.py
similarity index 95%
rename from cli.py
rename to demo/demo_cli.py
index 5f42f03..e95a1a7 100644
--- a/cli.py
+++ b/demo/demo_cli.py
@@ -2,7 +2,7 @@
 import sys
 from pathlib import Path
 
-demo_base_path = Path(__file__).parent / "demo"
+demo_base_path = Path(__file__).parent
 
 client_path = demo_base_path / "client.html"
 server_path = demo_base_path/ "server.py"
diff --git a/pyproject.toml b/pyproject.toml
index 7ec8fe0..7761f32 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -45,4 +45,4 @@ demo = [
 "Bug Tracker" = "https://github.com/catalystneuro/tqdm_publisher/issues"
 
 [project.gui-scripts]
-tqdm_publisher = "cli:main"
\ No newline at end of file
+tqdm_publisher = "demo.demo_cli:main"
\ No newline at end of file