coder
diff --git a/‎agent/agent.go‎
Lines changed: 4 additions & 2 deletions b/‎agent/agent.go‎
Lines changed: 4 additions & 2 deletions
@@ -1188,7 +1188,7 @@ func (a *agent) handleManifest(manifestOK *checkpoint) func(ctx context.Context,
 // createOrUpdateNetwork waits for the manifest to be set using manifestOK, then creates or updates
 // the tailnet using the information in the manifest
 func (a *agent) createOrUpdateNetwork(manifestOK, networkOK *checkpoint) func(context.Context, proto.DRPCAgentClient26) error {
- return func(ctx context.Context, _ proto.DRPCAgentClient26) (retErr error) {
+ return func(ctx context.Context, aAPI proto.DRPCAgentClient26) (retErr error) {
  if err := manifestOK.wait(ctx); err != nil {
  return xerrors.Errorf("no manifest: %w", err)
 		}
@@ -1208,6 +1208,7 @@ func (a *agent) createOrUpdateNetwork(manifestOK, networkOK *checkpoint) func(co
  // agent API.
  network, err = a.createTailnet(
  a.gracefulCtx,
+ aAPI,
  manifest.AgentID,
  manifest.DERPMap,
  manifest.DERPForceWebSockets,
@@ -1355,6 +1356,7 @@ func (a *agent) trackGoroutine(fn func()) error {
 
 func (a *agent) createTailnet(
  ctx context.Context,
+ aAPI proto.DRPCAgentClient26,
  agentID uuid.UUID,
  derpMap *tailcfg.DERPMap,
  derpForceWebSockets, disableDirectConnections bool,
@@ -1487,7 +1489,7 @@ func (a *agent) createTailnet(
 	}()
  if err = a.trackGoroutine(func() {
  defer apiListener.Close()
- apiHandler, closeAPIHAndler := a.apiHandler()
+ apiHandler, closeAPIHAndler := a.apiHandler(aAPI)
  defer func() {
  _ = closeAPIHAndler()
 		}()