diff --git a/pandasai/smart_datalake/__init__.py b/pandasai/smart_datalake/__init__.py
index 082ffa9c1..e16d2b7f8 100644
--- a/pandasai/smart_datalake/__init__.py
+++ b/pandasai/smart_datalake/__init__.py
@@ -129,8 +129,8 @@ def __init__(
             server_config=self._config.log_server,
         )
 
-    def set_instance_type(self, type: str):
-        self._instance = type
+    def set_instance_type(self, type_: str):
+        self._instance = type_
 
     def is_related_query(self, flag: bool):
         self._query_exec_tracker.set_related_query(flag)