diff --git a/packages/editor/tests/unit/layouts/workspace/Stage.spec.ts b/packages/editor/tests/unit/layouts/workspace/Stage.spec.ts
index a02c33a3..ab7a707e 100644
--- a/packages/editor/tests/unit/layouts/workspace/Stage.spec.ts
+++ b/packages/editor/tests/unit/layouts/workspace/Stage.spec.ts
@@ -64,6 +64,7 @@ describe('Stage.vue', () => {
       page,
       node: page,
       uiSelectMode: false,
+      customContentMenu: (menu: any) => menu,
     },
   });