diff --git a/adetailer/args.py b/adetailer/args.py
index a5f477d..e8ef28a 100644
--- a/adetailer/args.py
+++ b/adetailer/args.py
@@ -110,7 +110,7 @@ class ADetailerArgs(BaseModel, extra=Extra.forbid):
 def enable_check(*args: Any) -> bool:
     if not args:
         return False
-    a0 = args[0]
+    a0: bool | Mapping = args[0]
     ad_model = ALL_ARGS[0].attr
 
     if isinstance(a0, Mapping):
diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py
index f141a70..cc440a9 100644
--- a/scripts/!adetailer.py
+++ b/scripts/!adetailer.py
@@ -355,14 +355,10 @@ class AfterDetailerScript(scripts.Script):
             )
 
         all_inputs = [state] + w.tolist()
-        if is_img2img:
-            img2img_submit_button.click(
-                fn=on_generate_click, inputs=all_inputs, outputs=state, queue=False
-            )
-        else:
-            txt2img_submit_button.click(
-                fn=on_generate_click, inputs=all_inputs, outputs=state, queue=False
-            )
+        target_button = img2img_submit_button if is_img2img else txt2img_submit_button
+        target_button.click(
+            fn=on_generate_click, inputs=all_inputs, outputs=state, queue=False
+        )
 
         infotext_fields = [
             (getattr(w, attr), name + suffix(n)) for attr, name in ALL_ARGS