diff --git a/extensions/llama_prompts/script.py b/extensions/llama_prompts/script.py
index e45cd445..22c96f7c 100644
--- a/extensions/llama_prompts/script.py
+++ b/extensions/llama_prompts/script.py
@@ -11,7 +11,7 @@ def get_prompt_by_name(name):
         return df[df['Prompt name'] == name].iloc[0]['Prompt'].replace('\\n', '\n')
 
 def ui():
-    if not shared.args.chat or share.args.cai_chat:
+    if not shared.args.chat or shared.args.cai_chat:
         choices = ['None'] + list(df['Prompt name'])
 
         prompts_menu = gr.Dropdown(value=choices[0], choices=choices, label='Prompt')