diff --git a/packages/hooks/src/use-block-suite-page-references.ts b/packages/hooks/src/use-block-suite-page-references.ts index 8a46cde154..60c7d84028 100644 --- a/packages/hooks/src/use-block-suite-page-references.ts +++ b/packages/hooks/src/use-block-suite-page-references.ts @@ -7,8 +7,8 @@ import { useBlockSuiteWorkspacePage } from './use-block-suite-workspace-page'; const weakMap = new WeakMap>(); function getPageReferences(page: Page): string[] { // todo: is there a way to use page indexer to get all references? - return page - .getBlockByFlavour('affine:paragraph') + return ['affine:paragraph', 'affine:list', 'affine:database'] + .flatMap(f => page.getBlockByFlavour(f)) .flatMap(b => b.text?.toDelta()) .map(v => v?.attributes?.reference?.pageId) .filter(Boolean);