diff --git a/assets/js/src/application.js b/assets/js/src/application.js
index 9c236b7c4bd5dd820985f2277e326967b490bcc4..eb52848756674da6e62f8af044a4a9c13a34aceb 100644
--- a/assets/js/src/application.js
+++ b/assets/js/src/application.js
@@ -54,7 +54,7 @@
     })
 
     // Insert copy to clipboard button before .highlight
-    $('.highlight').each(function () {
+    $('figure.highlight, div.highlight').each(function () {
       var btnHtml = '<div class="bd-clipboard"><button class="btn-clipboard" title="Copy to clipboard">Copy</button></div>'
       $(this).before(btnHtml)
       $('.btn-clipboard')