diff --git a/assets/css/common/main.css b/assets/css/common/main.css
index eb2c4c84..82b61246 100644
--- a/assets/css/common/main.css
+++ b/assets/css/common/main.css
@@ -44,7 +44,7 @@ code {
     direction: ltr;
 }
 
-div.highlight {
+div.highlight, pre {
     position: relative;
 }
 
@@ -60,6 +60,7 @@ div.highlight {
     font-size: 14px;
 }
 
-div.highlight:hover .copy-code {
+div.highlight:hover .copy-code,
+pre:hover .copy-code {
     display: block;
 }
diff --git a/layouts/partials/footer.html b/layouts/partials/footer.html
index 1badd4a6..2c6511b8 100644
--- a/layouts/partials/footer.html
+++ b/layouts/partials/footer.html
@@ -120,7 +120,17 @@
             selection.removeRange(range);
         });
 
-        container.appendChild(copybutton);
+        if (container.classList.contains("highlight")) {
+            container.appendChild(copybutton);
+        } else if (container.parentNode.firstChild == container) {
+            // td containing LineNos
+        } else if (codeblock.parentNode.parentNode.parentNode.parentNode.parentNode.nodeName == "TABLE") {
+            // table containing LineNos and code
+            codeblock.parentNode.parentNode.parentNode.parentNode.parentNode.appendChild(copybutton);
+        } else {
+            // code blocks not having highlight as parent class
+            codeblock.parentNode.appendChild(copybutton);
+        }
     });
 </script>
 {{- end }}