diff --git a/databend_py/__init__.py b/databend_py/__init__.py
index 92fab7a..9d9777e 100644
--- a/databend_py/__init__.py
+++ b/databend_py/__init__.py
@@ -2,7 +2,7 @@
 from .connection import Connection
 from .datetypes import DatabendDataType
 
-VERSION = (0, 3, 9)
+VERSION = (0, 4, 1)
 __version__ = '.'.join(str(x) for x in VERSION)
 
 __all__ = ['Client', 'Connection', 'DatabendDataType']
diff --git a/databend_py/connection.py b/databend_py/connection.py
index b04fb2e..25c7b91 100644
--- a/databend_py/connection.py
+++ b/databend_py/connection.py
@@ -110,9 +110,11 @@ def query(self, statement):
         log.logger.debug(f"http sql: {statement}")
         query_sql = {'sql': statement, "string_fields": True}
         if self.client_session is not None and len(self.client_session) != 0:
+            if "database" not in self.client_session:
+                self.client_session = {"database": self.database}
             query_sql['session'] = self.client_session
         else:
-            self.client_session = {"db": self.database}
+            self.client_session = {"database": self.database}
             query_sql['session'] = self.client_session
         log.logger.debug(f"http headers {self.make_headers()}")
         response = requests.post(url,