diff --git a/docs/ai-coder/ai-gateway/clients/claude-code.md b/docs/ai-coder/ai-gateway/clients/claude-code.md
index 6680de6ebf..17b851d133 100644
--- a/docs/ai-coder/ai-gateway/clients/claude-code.md
+++ b/docs/ai-coder/ai-gateway/clients/claude-code.md
@@ -55,11 +55,11 @@ Template admins can pre-configure Claude Code for a seamless experience. Admins
 
 ```hcl
 module "claude-code" {
-  source          = "registry.coder.com/coder/claude-code/coder"
-  version         = "4.7.3"
-  agent_id        = coder_agent.main.id
-  workdir         = "/path/to/project"  # Set to your project directory
-  enable_aibridge = true
+  source            = "registry.coder.com/coder/claude-code/coder"
+  version           = "4.7.3"
+  agent_id          = coder_agent.main.id
+  workdir           = "/path/to/project"  # Set to your project directory
+  enable_ai_gateway = true
 }
 ```
 
@@ -76,14 +76,14 @@ resource "coder_ai_task" "task" {
 data "coder_task" "me" {}
 
 module "claude-code" {
-  source         = "registry.coder.com/coder/claude-code/coder"
-  version        = "4.7.3"
-  agent_id       = coder_agent.main.id
-  workdir        = "/path/to/project"  # Set to your project directory
-  ai_prompt      = data.coder_task.me.prompt
+  source            = "registry.coder.com/coder/claude-code/coder"
+  version           = "4.7.3"
+  agent_id          = coder_agent.main.id
+  workdir           = "/path/to/project"  # Set to your project directory
+  ai_prompt         = data.coder_task.me.prompt
 
   # Route through AI Gateway (AI Governance Add-On)
-  enable_aibridge = true
+  enable_ai_gateway = true
 }
 ```
 
diff --git a/docs/ai-coder/ai-gateway/clients/codex.md b/docs/ai-coder/ai-gateway/clients/codex.md
index 2c25521608..202524fa1b 100644
--- a/docs/ai-coder/ai-gateway/clients/codex.md
+++ b/docs/ai-coder/ai-gateway/clients/codex.md
@@ -91,11 +91,11 @@ If configuring within a Coder workspace, you can use the
 
 ```tf
 module "codex" {
-  source          = "registry.coder.com/coder-labs/codex/coder"
-  version         = "~> 4.1"
-  agent_id        = coder_agent.main.id
-  workdir         = "/path/to/project"  # Set to your project directory
-  enable_aibridge = true
+  source            = "registry.coder.com/coder-labs/codex/coder"
+  version           = "~> 4.1"
+  agent_id          = coder_agent.main.id
+  workdir           = "/path/to/project"  # Set to your project directory
+  enable_ai_gateway = true
 }
 ```