diff --git a/private_gpt/components/tabular/database_query_generator.py b/private_gpt/components/tabular/database_query_generator.py
index ec67d16f..eedf9a88 100644
--- a/private_gpt/components/tabular/database_query_generator.py
+++ b/private_gpt/components/tabular/database_query_generator.py
@@ -435,7 +435,9 @@ class DatabaseQueryGenerator:
         """
         try:
             conn = self._ensure_connected()
-            dialect = classify_dialect(self._engine.dialect.name if self._engine else None)
+            dialect = classify_dialect(
+                self._engine.dialect.name if self._engine else None
+            )
             if dialect == DatabaseDialect.DB2:
                 conn.execute(text("SELECT 1 FROM SYSIBM.SYSDUMMY1"))
             else: