diff --git a/src/types/index.d.ts b/src/types/index.d.ts
index 518bf1dca28..1a652c6dc19 100644
--- a/src/types/index.d.ts
+++ b/src/types/index.d.ts
@@ -516,7 +516,7 @@ export declare class Chart<
   buildOrUpdateScales(): void;
   buildOrUpdateControllers(): void;
   reset(): void;
-  update(mode?: UpdateMode): void;
+  update(mode?: UpdateMode | ((ctx: { datasetIndex: number }) => UpdateMode)): void;
   render(): void;
   draw(): void;