diff --git a/tests/test_references.py b/tests/test_references.py
index 323813d..585529e 100644
--- a/tests/test_references.py
+++ b/tests/test_references.py
@@ -116,7 +116,7 @@ def test_reference_implicit_with_code_inlinehilite_python() -> None:
         extensions={"pymdownx.inlinehilite": {"style_plain_text": "python"}, "pymdownx.highlight": {}},
         url_map={"pathlib.Path": "pathlib.html#Path"},
         source="This [`pathlib.Path`][].",
-        output='<p>This <a class="autorefs autorefs-internal" href="pathlib.html#Path"><code class="highlight">pathlib.Path</code></a>.</p>',
+        output='<p>This <a class="autorefs autorefs-internal" href="pathlib.html#Path"><code class="language-python highlight">pathlib.Path</code></a>.</p>',
     )