diff --git a/ai/models.py b/ai/models.py
index 7b58a02..0333b23 100644
--- a/ai/models.py
+++ b/ai/models.py
@@ -27,7 +27,7 @@ model_logic = None
 model_writer = None
 model_artist = None
 model_image = None
-logic_model_name = "models/gemini-1.5-pro"
+logic_model_name = "models/gemini-1.5-flash"
 writer_model_name = "models/gemini-1.5-flash"
 artist_model_name = "models/gemini-1.5-flash"
 image_model_name = None
diff --git a/story/writer.py b/story/writer.py
index aeb732a..170edd6 100644
--- a/story/writer.py
+++ b/story/writer.py
@@ -327,7 +327,7 @@ def write_chapter(chap, bp, folder, prev_sum, tracking=None, prev_content=None,
         utils.log("WRITER", f"⚠️ Failed Ch {chap['chapter_number']}: {e}")
         return f"## Chapter {chap['chapter_number']} Failed\n\nError: {e}"
 
-    max_attempts = 5
+    max_attempts = 2
     SCORE_AUTO_ACCEPT = 8
     SCORE_PASSING = 7
     SCORE_REWRITE_THRESHOLD = 6
@@ -378,11 +378,14 @@ def write_chapter(chap, bp, folder, prev_sum, tracking=None, prev_content=None,
             """
 
             try:
+                ai_models.model_logic.update("models/gemini-1.5-pro")
                 resp_rewrite = ai_models.model_logic.generate_content(full_rewrite_prompt)
                 utils.log_usage(folder, ai_models.model_logic.name, resp_rewrite.usage_metadata)
                 current_text = resp_rewrite.text
+                ai_models.model_logic.update("models/gemini-1.5-flash")
                 continue
             except Exception as e:
+                ai_models.model_logic.update("models/gemini-1.5-flash")
                 utils.log("WRITER", f"Full rewrite failed: {e}. Falling back to refinement.")
 
         utils.log("WRITER", f"  -> Refining Ch {chap['chapter_number']} based on feedback...")