diff --git a/api/controllers/console/datasets/rag_pipeline/rag_pipeline.py b/api/controllers/console/datasets/rag_pipeline/rag_pipeline.py
index 0d882e29e8..3be655a105 100644
--- a/api/controllers/console/datasets/rag_pipeline/rag_pipeline.py
+++ b/api/controllers/console/datasets/rag_pipeline/rag_pipeline.py
@@ -157,6 +157,6 @@ api.add_resource(
     "/rag/pipeline/customized/templates/<string:template_id>",
 )
 api.add_resource(
-    CustomizedPipelineTemplateApi,
+    PublishCustomizedPipelineTemplateApi,
     "/rag/pipeline/customized/templates/<string:template_id>/publish",
 )