diff --git a/libs/components/editor-core/src/render-block/Context.tsx b/libs/components/editor-core/src/render-block/Context.tsx
index d34f5d6801..e5c0197eee 100644
--- a/libs/components/editor-core/src/render-block/Context.tsx
+++ b/libs/components/editor-core/src/render-block/Context.tsx
@@ -27,7 +27,7 @@ export const BlockRenderProvider = ({
     );
 };
 
-const useBlockRender = () => {
+export const useBlockRender = () => {
     const { blockRender } = useContext(BlockRenderContext);
     return {
         BlockRender: blockRender,
diff --git a/libs/components/editor-core/src/render-block/WithTreeViewChildren.tsx b/libs/components/editor-core/src/render-block/WithTreeViewChildren.tsx
index 6dccd49ace..8c3642ef6d 100644
--- a/libs/components/editor-core/src/render-block/WithTreeViewChildren.tsx
+++ b/libs/components/editor-core/src/render-block/WithTreeViewChildren.tsx
@@ -7,7 +7,7 @@ import type {
 } from 'react';
 import { forwardRef } from 'react';
 import { CreateView } from '../editor';
-import { BlockRender } from './Context';
+import { useBlockRender } from './Context';
 import { NullBlockRender } from './RenderBlock';
 
 type WithChildrenConfig = {
@@ -60,6 +60,7 @@ export const withTreeViewChildren = (
 
     return (props: CreateView) => {
         const { block } = props;
+        const { BlockRender } = useBlockRender();
         const collapsed = block.getProperty('collapsed')?.value;
         const childrenIds = block.childrenIds;
         const showChildren =