diff --git a/index.d.ts b/index.d.ts
index c453a7e..9249e5d 100644
--- a/index.d.ts
+++ b/index.d.ts
@@ -9,7 +9,7 @@ interface ClientSetOptions {
 
 type EmbedCallback = (context: {
   parameters: { [k: string]: unknown }
-  userAgent: string
+  client: Client
 }) => Promise<string>
 
 type WidgetCallback = () => Promise<string>