diff --git a/pandasai/helpers/code_manager.py b/pandasai/helpers/code_manager.py
index b2fd0a661..415cc64f6 100644
--- a/pandasai/helpers/code_manager.py
+++ b/pandasai/helpers/code_manager.py
@@ -46,12 +46,12 @@ def __init__(
         if self._config.middlewares is not None:
             self.add_middlewares(*self._config.middlewares)
 
-    def add_middlewares(self, *middlewares: List[Middleware]):
+    def add_middlewares(self, *middlewares: Optional[Middleware]):
         """
         Add middlewares to PandasAI instance.
 
         Args:
-            *middlewares: A list of middlewares
+            *middlewares: Middlewares to be added
 
         """
         self._middlewares.extend(middlewares)
diff --git a/pandasai/smart_dataframe/__init__.py b/pandasai/smart_dataframe/__init__.py
index 7787a199b..f5af2d56f 100644
--- a/pandasai/smart_dataframe/__init__.py
+++ b/pandasai/smart_dataframe/__init__.py
@@ -177,12 +177,12 @@ def __setitem__(self, key, value):
     def __repr__(self):
         return self._df.__repr__()
 
-    def add_middlewares(self, *middlewares: List[Middleware]):
+    def add_middlewares(self, *middlewares: Optional[Middleware]):
         """
         Add middlewares to PandasAI instance.
 
         Args:
-            *middlewares: A list of middlewares
+            *middlewares: Middlewares to be added
 
         """
         self._dl.add_middlewares(*middlewares)
diff --git a/pandasai/smart_datalake/__init__.py b/pandasai/smart_datalake/__init__.py
index 0c4f6f3db..4650a7d94 100644
--- a/pandasai/smart_datalake/__init__.py
+++ b/pandasai/smart_datalake/__init__.py
@@ -169,12 +169,12 @@ def _load_llm(self, llm: LLM):
 
         self._llm = llm
 
-    def add_middlewares(self, *middlewares: List[Middleware]):
+    def add_middlewares(self, *middlewares: Optional[Middleware]):
         """
         Add middlewares to PandasAI instance.
 
         Args:
-            *middlewares: A list of middlewares
+            *middlewares: Middlewares to be added
         """
         self._code_manager.add_middlewares(*middlewares)