diff --git a/apps/application/flow/common.py b/apps/application/flow/common.py
index 65aaa399d6f..dd69e4cd6fe 100644
--- a/apps/application/flow/common.py
+++ b/apps/application/flow/common.py
@@ -242,7 +242,8 @@ def is_valid_start_node(self):
             raise AppApiException(500, _('There can only be one starting node'))
 
     def is_valid_model_params(self):
-        node_list = [node for node in self.nodes if (node.type == 'ai-chat-node' or node.type == 'question-node')]
+        node_list = [node for node in self.nodes if (
+                    node.type == 'ai-chat-node' or node.type == 'question-node' or node.type == 'parameter-extraction-node')]
         for node in node_list:
             model = QuerySet(Model).filter(id=node.properties.get('node_data', {}).get('model_id')).first()
             if model is None: