diff --git a/src/databricks/sqlalchemy/__init__.py b/src/databricks/sqlalchemy/__init__.py
index 2a17ac3e..2dde351e 100644
--- a/src/databricks/sqlalchemy/__init__.py
+++ b/src/databricks/sqlalchemy/__init__.py
@@ -1,4 +1,4 @@
 from databricks.sqlalchemy.base import DatabricksDialect
 from databricks.sqlalchemy._types import TINYINT, TIMESTAMP, TIMESTAMP_NTZ
 
-__all__ = ["TINYINT", "TIMESTAMP", "TIMESTAMP_NTZ"]
+__all__ = ["DatabricksDialect", "TINYINT", "TIMESTAMP", "TIMESTAMP_NTZ"]