diff --git a/assets/js/code-copy-v2.js b/assets/js/code-copy-v2.js
index d1fe2c7..023032a 100644
--- a/assets/js/code-copy-v2.js
+++ b/assets/js/code-copy-v2.js
@@ -1,6 +1,5 @@
 // Tightly coupled to `render-codeblock.html` for element targeting
-const copyToClipBoard = ((clipboard) => async (button) => {
-  const codeBlockId = button.getAttribute('data-id-codeblock');
+const copyToClipBoard = ((clipboard) => async (button, codeBlockId) => {
   const codeBlock = document
     .getElementById(codeBlockId)
     .getElementsByClassName('highlight')[0];
diff --git a/layouts/_default/_markup/render-codeblock.html b/layouts/_default/_markup/render-codeblock.html
index 657c92b..00350d8 100644
--- a/layouts/_default/_markup/render-codeblock.html
+++ b/layouts/_default/_markup/render-codeblock.html
@@ -6,7 +6,7 @@
 {{- if $isSingleLine  -}}
 <div class="code-block" data-mf="true" style="display: none;">
     <div class="code-container">
-        <button onclick="copyToClipBoard(this)" data-id-codeblock="{{ $codeBlockId }}" class="code-copy-button" type="button">Copy</button>
+        <button onclick="copyToClipBoard(this, {{ $codeBlockId }})" class="code-copy-button" type="button">Copy</button>
         <div class="highlight-v2 single-line" id="{{ $codeBlockId }}">
             {{ $result.Wrapped }}
         </div>
@@ -18,7 +18,7 @@
         <span class="code-type">{{ .Type }}</span>
     {{- end -}}
     <div class="code-container">
-        <button onclick="copyToClipBoard(this)" data-id-codeblock="{{ $codeBlockId }}" class="code-copy-button multi-line" type="button">Copy</button>
+        <button onclick="copyToClipBoard(this, {{ $codeBlockId }})" class="code-copy-button multi-line" type="button">Copy</button>
         <div class="highlight-v2" id="{{ $codeBlockId }}">
             {{ $result.Wrapped }}
         </div>
diff --git a/tests/src/__screenshots__/Mobile-Chrome/visual-regression.spec.js/example-site-screenshot.png b/tests/src/__screenshots__/Mobile-Chrome/visual-regression.spec.js/example-site-screenshot.png
index 90dd3e3..d640642 100644
Binary files a/tests/src/__screenshots__/Mobile-Chrome/visual-regression.spec.js/example-site-screenshot.png and b/tests/src/__screenshots__/Mobile-Chrome/visual-regression.spec.js/example-site-screenshot.png differ
diff --git a/tests/src/__screenshots__/firefox/visual-regression.spec.js/example-site-screenshot.png b/tests/src/__screenshots__/firefox/visual-regression.spec.js/example-site-screenshot.png
index 5e6eee0..f3d85a5 100644
Binary files a/tests/src/__screenshots__/firefox/visual-regression.spec.js/example-site-screenshot.png and b/tests/src/__screenshots__/firefox/visual-regression.spec.js/example-site-screenshot.png differ
diff --git a/tests/src/__screenshots__/webkit/visual-regression.spec.js/example-site-screenshot.png b/tests/src/__screenshots__/webkit/visual-regression.spec.js/example-site-screenshot.png
index ac91dd7..b8af744 100644
Binary files a/tests/src/__screenshots__/webkit/visual-regression.spec.js/example-site-screenshot.png and b/tests/src/__screenshots__/webkit/visual-regression.spec.js/example-site-screenshot.png differ