fix(editor): surface ref find element error (#9508)

toeverything · Jan 3, 2025 · ee5f13c · ee5f13c
1 parent 51b109e
commit ee5f13c
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/blocksuite/affine/block-surface-ref/src/surface-ref-block.ts b/blocksuite/affine/block-surface-ref/src/surface-ref-block.ts
@@ -341,7 +341,7 @@ export class SurfaceRefBlockComponent extends BlockComponent<SurfaceRefBlockMode
         .find(
           doc =>
             doc.getBlock(this.model.reference) ||
-            getSurfaceBlock(doc)!.getElementById(this.model.reference)
+            getSurfaceBlock(doc)?.getElementById(this.model.reference)
         );
 
       if (doc) {
@@ -355,11 +355,11 @@ export class SurfaceRefBlockComponent extends BlockComponent<SurfaceRefBlockMode
         ];
       }
 
-      if (doc && getSurfaceBlock(doc)) {
-        return [
-          getSurfaceBlock(doc)!.getElementById(this.model.reference),
-          doc.id,
-        ];
+      if (doc) {
+        const surfaceBlock = getSurfaceBlock(doc);
+        if (surfaceBlock) {
+          return [surfaceBlock.getElementById(this.model.reference), doc.id];
+        }
       }
 
       return [null, this.doc.id];