diff --git a/js/samples/dev-ui-gallery/prompts/dotprompt-hello.history.prompt b/js/samples/dev-ui-gallery/prompts/dotprompt-hello.history.prompt
index 5fe9e667bf..23050fae2f 100644
--- a/js/samples/dev-ui-gallery/prompts/dotprompt-hello.history.prompt
+++ b/js/samples/dev-ui-gallery/prompts/dotprompt-hello.history.prompt
@@ -5,16 +5,15 @@ config:
   temperature: 0.6
   topK: 16
   topP: 0.95
-  custom:
-    safetySettings:
-      - category: HARM_CATEGORY_HATE_SPEECH
-        threshold: BLOCK_ONLY_HIGH
-      - category: HARM_CATEGORY_DANGEROUS_CONTENT
-        threshold: BLOCK_ONLY_HIGH
-      - category: HARM_CATEGORY_HARASSMENT
-        threshold: BLOCK_ONLY_HIGH
-      - category: HARM_CATEGORY_SEXUALLY_EXPLICIT
-        threshold: BLOCK_ONLY_HIGH
+  safetySettings:
+    - category: HARM_CATEGORY_HATE_SPEECH
+      threshold: BLOCK_ONLY_HIGH
+    - category: HARM_CATEGORY_DANGEROUS_CONTENT
+      threshold: BLOCK_ONLY_HIGH
+    - category: HARM_CATEGORY_HARASSMENT
+      threshold: BLOCK_ONLY_HIGH
+    - category: HARM_CATEGORY_SEXUALLY_EXPLICIT
+      threshold: BLOCK_ONLY_HIGH
 input:
   schema:
     name: string
diff --git a/js/samples/dev-ui-gallery/prompts/dotprompt-hello.prompt b/js/samples/dev-ui-gallery/prompts/dotprompt-hello.prompt
index c40058a864..4f8921a638 100644
--- a/js/samples/dev-ui-gallery/prompts/dotprompt-hello.prompt
+++ b/js/samples/dev-ui-gallery/prompts/dotprompt-hello.prompt
@@ -7,16 +7,15 @@ config:
   topP: 0.95
   stopSequences:
     - STAWP!
-  custom:
-    safetySettings:
-      - category: HARM_CATEGORY_HATE_SPEECH
-        threshold: BLOCK_ONLY_HIGH
-      - category: HARM_CATEGORY_DANGEROUS_CONTENT
-        threshold: BLOCK_ONLY_HIGH
-      - category: HARM_CATEGORY_HARASSMENT
-        threshold: BLOCK_ONLY_HIGH
-      - category: HARM_CATEGORY_SEXUALLY_EXPLICIT
-        threshold: BLOCK_ONLY_HIGH
+  safetySettings:
+    - category: HARM_CATEGORY_HATE_SPEECH
+      threshold: BLOCK_ONLY_HIGH
+    - category: HARM_CATEGORY_DANGEROUS_CONTENT
+      threshold: BLOCK_ONLY_HIGH
+    - category: HARM_CATEGORY_HARASSMENT
+      threshold: BLOCK_ONLY_HIGH
+    - category: HARM_CATEGORY_SEXUALLY_EXPLICIT
+      threshold: BLOCK_ONLY_HIGH
 input:
   schema:
     name: string