From 11aaa0ac7be69d9724b79e35d9a79c3c03ec0bad Mon Sep 17 00:00:00 2001
From: asagi4 <130366179+asagi4@users.noreply.github.com>
Date: Mon, 9 Jun 2025 01:06:44 +0300
Subject: [PATCH] Fix indexing error

---
 prompt_control/prompts.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/prompt_control/prompts.py b/prompt_control/prompts.py
index 3d9472a..bc78030 100644
--- a/prompt_control/prompts.py
+++ b/prompt_control/prompts.py
@@ -592,7 +592,7 @@ def encode_prompt(clip, text, start_pct, end_pct, defaults, masks):
         if attention_couple:
             fill = base_cond[0][1].get("x-promptcontrol.fill")
             if not fill:
-                ensure_mask(base_cond)
+                ensure_mask(base_cond[0])
             # else, set_cond_attnmask will have the base mask fill any unspecified areas
             base_cond = set_cond_attnmask(
                 base_cond,