diff --git a/scripts/controlnet.py b/scripts/controlnet.py
index 72dbddb..9bea442 100644
--- a/scripts/controlnet.py
+++ b/scripts/controlnet.py
@@ -1216,7 +1216,12 @@ class Script(scripts.Script, metaclass=(
                     weight = param.weight
 
                 h, w, hr_y, hr_x = Script.get_target_dimensions(p)
-                pulid_mode = PuLIDMode(unit.pulid_mode) if isinstance(unit.pulid_mode, str) else unit.pulid_mode
+                # TODO: Fix all enum issue
+                if unit.pulid_mode == "PuLIDMode.FIDELITY":
+                    pulid_mode = PuLIDMode.FIDELITY
+                else:
+                    pulid_mode = PuLIDMode(unit.pulid_mode)
+
                 if pulid_mode == PuLIDMode.STYLE:
                     pulid_attn_setting = PULID_SETTING_STYLE
                 else: