diff --git a/pgml-cms/docs/api/sql-extension/pgml.embed.md b/pgml-cms/docs/api/sql-extension/pgml.embed.md
index b31c944b3..1f9b946b5 100644
--- a/pgml-cms/docs/api/sql-extension/pgml.embed.md
+++ b/pgml-cms/docs/api/sql-extension/pgml.embed.md
@@ -14,7 +14,7 @@ The `pgml.embed()` function generates [embeddings](/docs/use-cases/embeddings/)
 pgml.embed(
     transformer TEXT,
     "text" TEXT,
-    kwargs JSON
+    kwargs JSONB
 )
 ```