diff --git a/forms-flow-api/requirements.txt b/forms-flow-api/requirements.txt
index 6e947ccad5..5b0a9897ec 100644
--- a/forms-flow-api/requirements.txt
+++ b/forms-flow-api/requirements.txt
@@ -28,7 +28,7 @@ exceptiongroup==1.2.2
 flask-jwt-oidc==0.7.0
 flask-marshmallow==1.2.1
 flask-restx==1.3.0
-formsflow_api_utils  @ git+https://github.com/shuhaib-aot/forms-flow-ai.git@feature/FWF-3679-subflow-creation-updation#subdirectory=forms-flow-api-utils
+formsflow_api_utils  @ git+https://github.com/AOT-Technologies/forms-flow-ai.git@develop#subdirectory=forms-flow-api-utils
 gunicorn==23.0.0
 h11==0.14.0
 h2==4.1.0
diff --git a/forms-flow-api/requirements/prod.txt b/forms-flow-api/requirements/prod.txt
index 5e94b8f72d..dfa553abd2 100644
--- a/forms-flow-api/requirements/prod.txt
+++ b/forms-flow-api/requirements/prod.txt
@@ -17,4 +17,4 @@ markupsafe
 PyJWT
 redis
 lxml
-git+https://github.com/shuhaib-aot/forms-flow-ai.git@feature/FWF-3679-subflow-creation-updation#subdirectory=forms-flow-api-utils
\ No newline at end of file
+git+https://github.com/AOT-Technologies/forms-flow-ai.git@develop#subdirectory=forms-flow-api-utils
\ No newline at end of file
diff --git a/forms-flow-api/src/formsflow_api/models/form_process_mapper.py b/forms-flow-api/src/formsflow_api/models/form_process_mapper.py
index 6dbe5984b5..5ac71429d4 100644
--- a/forms-flow-api/src/formsflow_api/models/form_process_mapper.py
+++ b/forms-flow-api/src/formsflow_api/models/form_process_mapper.py
@@ -282,7 +282,7 @@ def find_all_active_by_formid(
         )
         query = cls.add_search_filter(query=query, search=search)
         query = cls.access_filter(query=query)
-        query = cls.add_sort_filter(
+        query = add_sort_filter(
             sort_by=sort_by,
             sort_order=sort_order,
             query=query,