diff --git a/docs/source/users/index.md b/docs/source/users/index.md
index b5e7066b5..cf5e1b7b6 100644
--- a/docs/source/users/index.md
+++ b/docs/source/users/index.md
@@ -943,9 +943,11 @@ Here is an example for configuring the `bedrock` provider for `ai21.j2-mid-v1` m
 ```json
 {
     "AiExtension": {
-        "bedrock:ai21.j2-mid-v1": {
-            "model_kwargs": {
-                "maxTokens": 200
+        "model_parameters": {
+            "bedrock:ai21.j2-mid-v1": {
+                "model_kwargs": {
+                    "maxTokens": 200
+                }
             }
         }
     }