diff --git a/api/core/llm_generator/llm_generator.py b/api/core/llm_generator/llm_generator.py
index 503f8e3e8..8c1d17168 100644
--- a/api/core/llm_generator/llm_generator.py
+++ b/api/core/llm_generator/llm_generator.py
@@ -532,7 +532,7 @@ class LLMGenerator:
             model=model_config.get("name", ""),
         )
         match node_type:
-            case "llm", "agent":
+            case "llm" | "agent":
                 system_prompt = LLM_MODIFY_PROMPT_SYSTEM
             case "code":
                 system_prompt = LLM_MODIFY_CODE_SYSTEM