diff --git a/packages/extension-code-block-lowlight/src/lowlight-plugin.ts b/packages/extension-code-block-lowlight/src/lowlight-plugin.ts
index 25b95556..3ff2351f 100644
--- a/packages/extension-code-block-lowlight/src/lowlight-plugin.ts
+++ b/packages/extension-code-block-lowlight/src/lowlight-plugin.ts
@@ -25,11 +25,6 @@ function parseNodes(nodes: any[], className: string[] = []): { text: string, cla
     .flat()
 }
 
-function getHighlightNodes(result: any) {
-  // `.value` for lowlight v1, `.children` for lowlight v2
-  return result.value || result.children || []
-}
-
 function getDecorations({ doc, name, lowlight }: { doc: ProsemirrorNode, name: string, lowlight: any }) {
   const decorations: Decoration[] = []
 
@@ -57,6 +52,11 @@ function getDecorations({ doc, name, lowlight }: { doc: ProsemirrorNode, name: s
       })
     })
 
+  function getHighlightNodes(result) {
+    // `.value` for lowlight v1, `.children` for lowlight v2
+    return result.value || result.children || []
+  }
+
   return DecorationSet.create(doc, decorations)
 }