diff --git a/pyTigerGraph/ai/ai.py b/pyTigerGraph/ai/ai.py
index e2717f68..614f365e 100644
--- a/pyTigerGraph/ai/ai.py
+++ b/pyTigerGraph/ai/ai.py
@@ -41,7 +41,7 @@ def registerCustomQuery(self, function_header: str, description: str, docstring:
             "docstring": docstring,
             "param_types": param_types
         }
-        url = self.nlqs_host+"/"+self.conn.graphname+"/registercustomquery"
+        url = self.nlqs_host+"/"+self.conn.graphname+"/register_docs"
         return self.conn._req("POST", url, authMode="pwd", data = data, jsonData=True, resKey=None)
 
     def retrieveDocs(self, query:str, top_k:int = 3):
@@ -58,7 +58,7 @@ def retrieveDocs(self, query:str, top_k:int = 3):
             "query": query
         }
 
-        url = self.nlqs_host+"/"+self.conn.graphname+"/retrievedocs?top_k="+str(top_k)
+        url = self.nlqs_host+"/"+self.conn.graphname+"/retrieve_docs?top_k="+str(top_k)
         return self.conn._req("POST", url, authMode="pwd", data = data, jsonData=True, resKey=None, skipCheck=True) 
 
     def query(self, query):