diff --git a/upsonic/remote/on_prem.py b/upsonic/remote/on_prem.py
index a72d60a..ebcfe80 100644
--- a/upsonic/remote/on_prem.py
+++ b/upsonic/remote/on_prem.py
@@ -5,7 +5,7 @@
 import ast
 from functools import wraps
 
-
+from typing import List
 
 import pickle
 import os
@@ -363,7 +363,7 @@ def install_package(self, package:str) -> None:
                     self._log(f"Installing {package} to default_dir")
                 pip(["install", package])
 
-    def extract_the_requirements(self, key:str)-> list[str]:
+    def extract_the_requirements(self, key:str)-> List[str]:
         the_requirements = self.get_requirements(key)
         elements = []
         for each in the_requirements.split(","):
@@ -1437,7 +1437,7 @@ def check_function(self, value, print_prompts=False,model: Optional[str] = None)
             return True
         return similarity_explanation
 
-    def search_by_documentation(self, question:str) -> list[str]:
+    def search_by_documentation(self, question:str) -> List[str]:
         data = {"question": question}
         response = self._send_request("POST", "/search_by_documentation", data)
         result = []
@@ -1445,7 +1445,7 @@ def search_by_documentation(self, question:str) -> list[str]:
             result.append(i[0])
         return result
 
-    def search(self, question:str) -> list[str]:
+    def search(self, question:str) -> List[str]:
         return self.search_by_documentation(question)
 
     def get_default_ai_model(self) -> str: