diff --git a/shared/editor/extensions/Prism.ts b/shared/editor/extensions/Prism.ts
index b7b32046d..31c9e969e 100644
--- a/shared/editor/extensions/Prism.ts
+++ b/shared/editor/extensions/Prism.ts
@@ -74,9 +74,10 @@ function getDecorations({
   lineNumbers?: boolean;
 }) {
   const decorations: Decoration[] = [];
-  const blocks: { node: Node; pos: number }[] = findBlockNodes(doc).filter(
-    (item) => item.node.type.name === name
-  );
+  const blocks: { node: Node; pos: number }[] = findBlockNodes(
+    doc,
+    true
+  ).filter((item) => item.node.type.name === name);
 
   function parseNodes(
     nodes: refractor.RefractorNode[],
diff --git a/shared/editor/queries/findChildren.ts b/shared/editor/queries/findChildren.ts
index da2717e2b..c3757a1ae 100644
--- a/shared/editor/queries/findChildren.ts
+++ b/shared/editor/queries/findChildren.ts
@@ -50,8 +50,9 @@ export function findChildren(
  * are blocks.
  *
  * @param node The node to iterate over
+ * @param descend Whether to descend into a node
  * @returns Child nodes that are blocks
  */
-export function findBlockNodes(node: Node): NodeWithPos[] {
-  return findChildren(node, (child) => child.isBlock);
+export function findBlockNodes(node: Node, descend = false): NodeWithPos[] {
+  return findChildren(node, (child) => child.isBlock, descend);
 }