diff --git a/src/plugin-sdk/provider-model-shared.test.ts b/src/plugin-sdk/provider-model-shared.test.ts
index 7d5b3b5b28f..1bc811f2d31 100644
--- a/src/plugin-sdk/provider-model-shared.test.ts
+++ b/src/plugin-sdk/provider-model-shared.test.ts
@@ -20,7 +20,7 @@ describe("buildProviderReplayFamilyHooks", () => {
     });
   });
 
-  it("maps google-gemini replay families", () => {
+  it("maps google-gemini replay families", async () => {
     const hooks = buildProviderReplayFamilyHooks({
       family: "google-gemini",
     });
@@ -37,7 +37,7 @@ describe("buildProviderReplayFamilyHooks", () => {
       allowSyntheticToolResults: true,
     });
 
-    const sanitized = hooks.sanitizeReplayHistory?.({
+    const sanitized = await hooks.sanitizeReplayHistory?.({
       provider: "google",
       modelApi: "google-generative-ai",
       modelId: "gemini-3.1-pro-preview",