diff --git a/app/src/protyle/wysiwyg/getBlock.ts b/app/src/protyle/wysiwyg/getBlock.ts
index 67a9cc300..53242fa46 100644
--- a/app/src/protyle/wysiwyg/getBlock.ts
+++ b/app/src/protyle/wysiwyg/getBlock.ts
@@ -97,7 +97,7 @@ export const getContenteditableElement = (element: Element): Element => {
         return blockElement.firstElementChild;
     } else if ("NodeTable" === type) {
         return blockElement.querySelector("table");
-    } else if ("NodeCodeBlock" === type) {
+    } else if ("NodeCodeBlock" === type && blockElement.classList.contains("code-block")) {
         return blockElement.querySelector(".hljs").lastElementChild;
     } else if (["NodeBlockQueryEmbed", "NodeMathBlock", "NodeHTMLBlock"].includes(type)) {
         return element;