diff --git a/tests/schema/specs/links.spec.ts b/tests/schema/specs/links.spec.ts
index f1196bc00..2262cb2b1 100644
--- a/tests/schema/specs/links.spec.ts
+++ b/tests/schema/specs/links.spec.ts
@@ -29,7 +29,7 @@ test('Links', async () => {
             {
               "__typename": "BlockMarkup",
               "markup": "
-    <p><a href="/sites/default/files/2023-04/document_docx.docx" data-type="Media: Document" data-id="[id]">link to file</a></p>
+    <p><a href="/sites/default/files/2023-04/document_docx.docx" data-type="Media: Document" data-id="[id]" data-entity-type="media">link to file</a></p>
 
     <p><a href="/en/privacy" data-type="Content: Basic page" data-id="[id]" data-entity-type="node">link to page</a></p>
     ",