diff --git a/blocksuite/affine/blocks/database/src/database-block.ts b/blocksuite/affine/blocks/database/src/database-block.ts index 45c763d80..952a76e39 100644 --- a/blocksuite/affine/blocks/database/src/database-block.ts +++ b/blocksuite/affine/blocks/database/src/database-block.ts @@ -372,6 +372,7 @@ export class DatabaseBlockComponent extends CaptionedBlockComponent { if ( !this.std @@ -379,7 +380,8 @@ export class DatabaseBlockComponent extends CaptionedBlockComponent`, accent: 'warning', + onClose: () => { + notifyClosed = true; + }, }); } - this.removeEventListener('click', handler); } }; - this.addEventListener('click', handler); - this.disposables.add(() => { - this.removeEventListener('click', handler); - }); + this.disposables.addFromEvent(this, 'click', handler); } private readonly dataViewRootLogic = lazy(