diff --git a/types/index.d.ts b/types/index.d.ts
index 37ce545..e46f3fa 100644
--- a/types/index.d.ts
+++ b/types/index.d.ts
@@ -144,7 +144,7 @@ interface Toasted {
    * @param message
    * @param options
    */
-  register (name: string, message: string, options?: ToastOptions): void
+  register (name: string, message: string | ((payload: any) => string), options?: ToastOptions): void
 
   /**
    * Clear all toasts