Registering fault handlers on TMDS server start up

mye956 · Oct 3, 2024 · 492f265 · 492f265
1 parent cde7709
commit 492f265
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/agent/handlers/task_server_setup.go b/agent/handlers/task_server_setup.go
@@ -96,9 +96,8 @@ func taskServerSetup(
 	agentAPIV1HandlersSetup(muxRouter, state, credentialsManager, cluster, tmdsAgentState,
 		taskProtectionClientFactory, metricsFactory)
 
-	// TODO: Future PR to pass in TMDS server router once all of the handlers have been implemented.
 	execWrapper := execwrapper.NewExec()
-	registerFaultHandlers(nil, tmdsAgentState, metricsFactory, execWrapper)
+	registerFaultHandlers(muxRouter, tmdsAgentState, metricsFactory, execWrapper)
 
 	return tmds.NewServer(auditLogger,
 		tmds.WithHandler(muxRouter),