diff --git a/modules/ui.py b/modules/ui.py
index ca45a444..9700d297 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -532,6 +532,8 @@ def setup_auto_save():
         'include_past_attachments',
 
         # Image generation tab (ui_image_generation.py)
+        'image_prompt',
+        'image_neg_prompt',
         'image_width',
         'image_height',
         'image_aspect_ratio',