diff --git a/registry/coder/modules/claude-code/main.tf b/registry/coder/modules/claude-code/main.tf
index bd160ded..3b1f7d66 100644
--- a/registry/coder/modules/claude-code/main.tf
+++ b/registry/coder/modules/claude-code/main.tf
@@ -338,8 +338,9 @@ locals {
 }
 
 module "agentapi" {
-  source  = "registry.coder.com/coder/agentapi/coder"
-  version = "2.0.0"
+  # 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"
 
   agent_id             = var.agent_id
   web_app_slug         = local.app_slug