diff --git a/tests/CodeBlocksTest.php b/tests/CodeBlocksTest.php
index a2aaad5..797648f 100644
--- a/tests/CodeBlocksTest.php
+++ b/tests/CodeBlocksTest.php
@@ -31,8 +31,8 @@ final class CodeBlocksTest extends TestCase {
 ```
 		";
 		$target = "
-<pre data-lang=\"php\">
-	<code>
+<pre data-lang=\"php\" class=\"language-php\">
+	<code data-lang=\"php\" class=\"language-php\">
 		public function testLanguageAnnotationParsesCorrectly() : bool {
 			return true;
 		}