diff --git a/pkg/cmd/agent-task/capi/client.go b/pkg/cmd/agent-task/capi/client.go
index 1e9cad3c8..ecec9a024 100644
--- a/pkg/cmd/agent-task/capi/client.go
+++ b/pkg/cmd/agent-task/capi/client.go
@@ -7,6 +7,8 @@ import (
 	"github.com/cli/cli/v2/internal/gh"
 )
 
+//go:generate moq -rm -out client_mock.go . CapiClient
+
 const baseCAPIURL = "https://api.githubcopilot.com"
 const capiHost = "api.githubcopilot.com"