diff --git a/openswarm.config.mjs b/openswarm.config.mjs
index 625b5e9..a4722b9 100644
--- a/openswarm.config.mjs
+++ b/openswarm.config.mjs
@@ -13,7 +13,7 @@ export const product = {
   mdnsDomain: "openswarm.local",
   starterRepo: "VRSEN/OpenSwarm",
   starterFolder: "openswarm",
-  entryFiles: "swarm.py",
+  entryFiles: "swarm.py,agency.py",
 }
 
 export const productTuiLogoLeft = [
diff --git a/openswarm.product-env.json b/openswarm.product-env.json
index 44b9f92..bdc2e04 100644
--- a/openswarm.product-env.json
+++ b/openswarm.product-env.json
@@ -9,7 +9,7 @@
   "AGENTSWARM_PRODUCT_MDNS_DOMAIN": "openswarm.local",
   "AGENTSWARM_PRODUCT_STARTER_REPO": "VRSEN/OpenSwarm",
   "AGENTSWARM_PRODUCT_STARTER_FOLDER": "openswarm",
-  "AGENTSWARM_PRODUCT_ENTRY_FILES": "swarm.py",
+  "AGENTSWARM_PRODUCT_ENTRY_FILES": "swarm.py,agency.py",
   "AGENTSWARM_PRODUCT_SKIP_POST_AUTH_MODEL_SELECTION": "true",
   "AGENTSWARM_PRODUCT_TUI_LOGO_LEFT": "[\"                                    \",\" ██████╗ ██████╗ ███████╗███╗   ██╗\",\"██╔═══██╗██╔══██╗██╔════╝████╗  ██║\",\"██║   ██║██████╔╝█████╗  ██╔██╗ ██║\",\"██║   ██║██╔═══╝ ██╔══╝  ██║╚██╗██║\",\"╚██████╔╝██║     ███████╗██║ ╚████║\",\" ╚═════╝ ╚═╝     ╚══════╝╚═╝  ╚═══╝\"]",
   "AGENTSWARM_PRODUCT_TUI_LOGO_RIGHT": "[\"\",\"███████╗██╗    ██╗ █████╗ ██████╗ ███╗   ███╗\",\"██╔════╝██║    ██║██╔══██╗██╔══██╗████╗ ████║\",\"███████╗██║ █╗ ██║███████║██████╔╝██╔████╔██║\",\"╚════██║██║███╗██║██╔══██╗██╔══██╗██║╚██╔╝██║\",\"███████║╚███╔███╔╝██║  ██║██║  ██║██║ ╚═╝ ██║\",\"╚══════╝ ╚══╝╚══╝ ╚═╝  ╚═╝╚═╝  ╚═╝╚═╝     ╚═╝\"]",
diff --git a/scripts/smoke-run-mode.py b/scripts/smoke-run-mode.py
index f978c0a..db7f304 100644
--- a/scripts/smoke-run-mode.py
+++ b/scripts/smoke-run-mode.py
@@ -517,8 +517,8 @@ def main() -> int:
             ),
             encoding="utf-8",
         )
-        create_local_openswarm_project(package_dir, generic_dir)
-        plain = run_tui_smoke(launcher, package_dir, generic_dir, root, env, args.check, args.prompt, args.expect, args.timeout)
+        project_dir = create_local_openswarm_project(package_dir, generic_dir)
+        plain = run_tui_smoke(launcher, package_dir, project_dir, root, env, args.check, args.prompt, args.expect, args.timeout)
         if "Agency Swarm Default" not in plain:
             raise RuntimeError("Smoke response was seen, but Agency Swarm Run mode was not detected")
         if args.check in {"agents", "all"}: