diff --git a/cmd/internal/tools_file_test.go b/cmd/internal/tools_file_test.go
index 511ec3e5ece1..dbb29c55e250 100644
--- a/cmd/internal/tools_file_test.go
+++ b/cmd/internal/tools_file_test.go
@@ -1788,8 +1788,8 @@ func TestPrebuiltTools(t *testing.T) {
 			name: "dataplex prebuilt tools",
 			in:   dataplex_config,
 			wantToolset: server.ToolsetConfigs{
-				"dataplex_tools": tools.ToolsetConfig{
-					Name:      "dataplex_tools",
+				"discovery": tools.ToolsetConfig{
+					Name:      "discovery",
 					ToolNames: []string{"search_entries", "lookup_entry", "search_aspect_types"},
 				},
 			},
diff --git a/internal/prebuiltconfigs/tools/dataplex.yaml b/internal/prebuiltconfigs/tools/dataplex.yaml
index f98c105ce9cc..c54b8ae4d38e 100644
--- a/internal/prebuiltconfigs/tools/dataplex.yaml
+++ b/internal/prebuiltconfigs/tools/dataplex.yaml
@@ -32,7 +32,7 @@ tools:
     description: Use this tool to find aspect types relevant to the query.
 
 toolsets:
-  dataplex_tools:
+  discovery:
     - search_entries
     - lookup_entry
     - search_aspect_types