diff --git a/ai_diffusion/custom_workflow.py b/ai_diffusion/custom_workflow.py
index afa38fa81..f090b45c1 100644
--- a/ai_diffusion/custom_workflow.py
+++ b/ai_diffusion/custom_workflow.py
@@ -288,7 +288,8 @@ def compare(a: str, b: str):
 def workflow_parameters(w: ComfyWorkflow):
     text_types = ("text", "prompt (positive)", "prompt (negative)")
     for node in w:
-        match (node.type, node.input("type", "")):
+        param_type = node.input("type", "") if node.type == "ETN_Parameter" else ""
+        match (node.type, param_type):
             case ("ETN_KritaStyle", _):
                 name = node.input("name", "Style")
                 yield CustomParam(ParamKind.style, name, node.input("sampler_preset", "auto"))