diff --git a/registry/coder/modules/claude-code/main.tf b/registry/coder/modules/claude-code/main.tf
index 3b1f7d66..bd160ded 100644
--- a/registry/coder/modules/claude-code/main.tf
+++ b/registry/coder/modules/claude-code/main.tf
@@ -338,9 +338,8 @@ locals {
 }
 
 module "agentapi" {
-  # source  = "registry.coder.com/coder/agentapi/coder"
-  # version = "2.0.0"
-  source = "github.com/coder/registry//registry/coder/modules/agentapi?ref=cj%2Fagentapi%2Fimprove-wait-for-start"
+  source  = "registry.coder.com/coder/agentapi/coder"
+  version = "2.0.0"
 
   agent_id             = var.agent_id
   web_app_slug         = local.app_slug