From 214a679f25d85e4fb83f5801ae135a021a509c68 Mon Sep 17 00:00:00 2001
From: William Conti <william.conti@datadoghq.com>
Date: Thu, 19 Sep 2024 11:04:59 -0400
Subject: [PATCH] fix test

---
 .../dd-trace/test/datastreams/schemas/schema_builder.spec.js    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/dd-trace/test/datastreams/schemas/schema_builder.spec.js b/packages/dd-trace/test/datastreams/schemas/schema_builder.spec.js
index db602ef83aa..134724b593a 100644
--- a/packages/dd-trace/test/datastreams/schemas/schema_builder.spec.js
+++ b/packages/dd-trace/test/datastreams/schemas/schema_builder.spec.js
@@ -24,7 +24,7 @@ describe('SchemaBuilder', () => {
     const shouldExtractAddress = builder.shouldExtractSchema('address', 1)
     const shouldExtractPerson2 = builder.shouldExtractSchema('person', 0)
     const shouldExtractTooDeep = builder.shouldExtractSchema('city', 11)
-    const schema = builder.build()
+    const schema = SchemaBuilder.getSchemaDefinition(builder.build())
 
     const expectedSchema = {
       components: {