diff --git a/extensions/feishu/src/setup-surface.test.ts b/extensions/feishu/src/setup-surface.test.ts
index 394f4cc36e1..d68ee795abc 100644
--- a/extensions/feishu/src/setup-surface.test.ts
+++ b/extensions/feishu/src/setup-surface.test.ts
@@ -210,16 +210,11 @@ describe("feishu setup wizard", () => {
     });
 
     const setupWizard = feishuPlugin.setupWizard;
-    if (
-      !setupWizard ||
-      !("finalize" in setupWizard) ||
-      typeof setupWizard.finalize !== "function"
-    ) {
-      throw new Error("Expected Feishu setup wizard finalize hook");
+    if (!setupWizard || !("finalize" in setupWizard) || !setupWizard.finalize) {
+      throw new Error("feishu setupWizard.finalize unavailable");
     }
 
-    const result = await runSetupWizardFinalize({
-      finalize: setupWizard.finalize,
+    const result = await setupWizard.finalize({
       cfg: {
         channels: {
           feishu: {