diff --git a/api/controllers/service_api/app/chatflow_memory.py b/api/controllers/service_api/app/chatflow_memory.py
index 10db3231f2..c3e9e87c0a 100644
--- a/api/controllers/service_api/app/chatflow_memory.py
+++ b/api/controllers/service_api/app/chatflow_memory.py
@@ -1,4 +1,4 @@
-from flask_restful import Resource, reqparse
+from flask_restx import Resource, reqparse
 from sqlalchemy.orm import Session
 
 from controllers.service_api import api
diff --git a/api/controllers/web/chatflow_memory.py b/api/controllers/web/chatflow_memory.py
index 078040b204..d8fcd49b53 100644
--- a/api/controllers/web/chatflow_memory.py
+++ b/api/controllers/web/chatflow_memory.py
@@ -1,4 +1,4 @@
-from flask_restful import reqparse
+from flask_restx import reqparse
 from sqlalchemy.orm.session import Session
 
 from controllers.web import api