diff --git a/codemcp/agno.py b/codemcp/agno.py
index 89a090a..665dfac 100644
--- a/codemcp/agno.py
+++ b/codemcp/agno.py
@@ -62,9 +62,7 @@ async def serve_playground_app_async(
 
 
 async def main():
-    async with MCPTools(
-        f"{sys.executable()} -m codemcp.hot_reload_entry"
-    ) as codemcp:
+    async with MCPTools(f"{sys.executable()} -m codemcp.hot_reload_entry") as codemcp:
         # TODO: cli-ify the model
         agent = Agent(
             model=Claude(id="claude-3-7-sonnet-20250219"),
diff --git a/pyproject.toml b/pyproject.toml
index f1c8631..d55d3dd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -43,6 +43,9 @@ codemcp-multi = "codemcp.multi_entry:main"
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
+[tool.hatch.build.targets.wheel]
+include = ["codemcp/templates/**/*"]
+
 [tool.uv]
 
 [tool.ruff]