diff --git a/iz_helpers/ui.py b/iz_helpers/ui.py
index 0b7b998..51ce9d2 100644
--- a/iz_helpers/ui.py
+++ b/iz_helpers/ui.py
@@ -50,7 +50,7 @@ def on_ui_tabs():
                     jpr = readJsonPrompt(pr, True)
 
                     main_common_prompt_pre = gr.Textbox(
-                        value=jpr["commonPromptPrefix"], label="Common Prompt Prefix"
+                        value=jpr["prePrompt"], label="Common Prompt Prefix"
                     )
 
                     main_prompts = gr.Dataframe(
@@ -64,7 +64,7 @@ def on_ui_tabs():
                     )
 
                     main_common_prompt_suf = gr.Textbox(
-                        value=jpr["commonPromptSuffix"], label="Common Prompt Suffix"
+                        value=jpr["postPrompt"], label="Common Prompt Suffix"
                     )
 
                     main_negative_prompt = gr.Textbox(