diff --git a/public/src/uncloseai-embed-modal.js b/public/src/uncloseai-embed-modal.js
index 7ce28a4..98a01ae 100644
--- a/public/src/uncloseai-embed-modal.js
+++ b/public/src/uncloseai-embed-modal.js
@@ -97,8 +97,8 @@ function addCodeBlockCopyButtons(element) {
 	codeBlocks.forEach((codeBlock) => {
 		const pre = codeBlock.parentElement;
 		if (pre.tagName.toLowerCase() === "pre") {
-			// Check if copy button already exists to avoid duplicates
-			if (pre.querySelector('.code-copy-btn')) {
+			// Check if buttons already exist to avoid duplicates
+			if (pre.querySelector('.uncloseai-code-copy-btn')) {
 				return;
 			}