diff --git a/blocksuite/affine/blocks/callout/src/callout-keymap.ts b/blocksuite/affine/blocks/callout/src/callout-keymap.ts
index eac9b65233..9e7c84fe22 100644
--- a/blocksuite/affine/blocks/callout/src/callout-keymap.ts
+++ b/blocksuite/affine/blocks/callout/src/callout-keymap.ts
@@ -29,8 +29,9 @@ export const CalloutKeymapExtension = KeymapExtension(std => {
         if (!parent || !matchModels(parent, [CalloutBlockModel])) {
           return false;
         }
-        calloutBlock = std.store.getBlock(parent.id);
-        if (!calloutBlock) return false;
+        const parentBlock = std.store.getBlock(parent.id);
+        if (!parentBlock) return false;
+        calloutBlock = parentBlock;
       }
 
       ctx.get('keyboardState').raw.preventDefault();