diff --git a/modules/control/units/controlnet.py b/modules/control/units/controlnet.py
index 8efbce5a5..96175a33b 100644
--- a/modules/control/units/controlnet.py
+++ b/modules/control/units/controlnet.py
@@ -179,7 +179,7 @@ def api_list_models(model_type: str | None = None):
         model_list += list(predefined_qwen)
     if model_type == 'hunyuandit' or model_type == 'all':
         model_list += list(predefined_hunyuandit)
-    if model_type == 'zimage':
+    if model_type == 'zimage' or model_type == 'all':
         model_list += list(predefined_zimage)
     model_list += sorted(find_models())
     return model_list
diff --git a/modules/control/units/lite_model.py b/modules/control/units/lite_model.py
index ef6d444f9..3095d7cfe 100644
--- a/modules/control/units/lite_model.py
+++ b/modules/control/units/lite_model.py
@@ -171,7 +171,7 @@ class ControlNetLLLite(torch.nn.Module): # pylint: disable=abstract-method
                 mapped_block, mapped_number = map_down_lllite_to_unet[int(block)]
                 b = model.down_blocks[mapped_block].attentions[int(mapped_number)].transformer_blocks[int(block_number)]
             elif root == 'output':
-                pass # not implemented
+                continue # not implemented
             else:
                 b = model.mid_block.attentions[0].transformer_blocks[int(block_number)]
             b = getattr(b, attn_name, None)