diff --git a/test/json/validation/custom-warnings.test.ts b/test/json/validation/custom-warnings.test.ts
index 261bc3f3..51465512 100644
--- a/test/json/validation/custom-warnings.test.ts
+++ b/test/json/validation/custom-warnings.test.ts
@@ -360,8 +360,8 @@ describe("custom-warnings.json - vault and market existence in Morpho API", () =
   ): Promise<{ uniqueKey: string } | null> {
     const query = `
       query MarketByUniqueKey($uniqueKey: String!, $chainId: Int!) {
-        marketByUniqueKey(uniqueKey: $uniqueKey, chainId: $chainId) {
-          uniqueKey
+        marketByUniqueKey: marketById(marketId: $uniqueKey, chainId: $chainId) {
+          uniqueKey: marketId
         }
       }
     `;