diff --git a/tests/data/html/code_block.html b/tests/data/html/code_block.html
index 9d48b41..18ed98d 100644
--- a/tests/data/html/code_block.html
+++ b/tests/data/html/code_block.html
@@ -1,4 +1,4 @@
-<div id="b-NzgT2MAhuSTnpFyo-E4"><pre><code data-lang="python"># Python Program to find the L.C.M. of two input number
+<div class="code-block-wrapper" id="b-NzgT2MAhuSTnpFyo-E4"><pre><code data-lang="python"># Python Program to find the L.C.M. of two input number
 
 def compute_lcm(x, y):
 
@@ -19,7 +19,7 @@
 num1 = 54
 num2 = 24
 
-print(&quot;The L.C.M. is&quot;, compute_lcm(num1, num2))</code></pre></div><div><pre><code># Python Program to find the L.C.M. of two input number
+print(&quot;The L.C.M. is&quot;, compute_lcm(num1, num2))</code></pre></div><div class="code-block-wrapper"><pre><code># Python Program to find the L.C.M. of two input number
 
 def compute_lcm(x, y):
 
diff --git a/tiptapy/templates/codeBlock.html b/tiptapy/templates/codeBlock.html
index 8313a53..2f3e046 100644
--- a/tiptapy/templates/codeBlock.html
+++ b/tiptapy/templates/codeBlock.html
@@ -2,7 +2,7 @@
 {%- if node.content -%}
   {%- set text = node.content[0].text -%}
   {%- if text -%}
-    <div{{ extract_tag_attrs(node) }}><pre>
+    <div class="code-block-wrapper" {{ extract_tag_attrs(node) }}><pre>
       {%- if language -%}
         <code data-lang="{{language}}">{{text}}</code>
       {%- else -%}
diff --git a/tiptapy/templates/code_block.html b/tiptapy/templates/code_block.html
index 8313a53..13f24ae 100644
--- a/tiptapy/templates/code_block.html
+++ b/tiptapy/templates/code_block.html
@@ -2,7 +2,7 @@
 {%- if node.content -%}
   {%- set text = node.content[0].text -%}
   {%- if text -%}
-    <div{{ extract_tag_attrs(node) }}><pre>
+    <div class="code-block-wrapper"{{ extract_tag_attrs(node) }}><pre>
       {%- if language -%}
         <code data-lang="{{language}}">{{text}}</code>
       {%- else -%}