diff --git a/tests/McpResponse.test.ts b/tests/McpResponse.test.ts
index 1a89f981..8cb48fc4 100644
--- a/tests/McpResponse.test.ts
+++ b/tests/McpResponse.test.ts
@@ -162,7 +162,7 @@ Emulating: 4x slowdown`,
       });
       await dialogPromise;
       const result = await response.handle('test', context);
-      context.getDialog()?.dismiss();
+      await context.getDialog()?.dismiss();
       assert.strictEqual(
         result[0].text,
         `# test response