diff --git a/pyproject.toml b/pyproject.toml
index 242d655..2cd8cc2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,12 +22,16 @@ dev = [
 ]
 
 [project.scripts]
-af-tcp-client = "af_server_client.cli:main"
+af-tcp-client = "af_client.cli:main"
 
 [build-system]
 requires = ["setuptools>=68.0.0", "wheel"]
 build-backend = "setuptools.build_meta"
 
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["af_client*"]
+
 [tool.ruff]
 line-length = 100
 target-version = "py310"