diff --git a/setup.py b/setup.py
index 66f99ac139..69a0747465 100644
--- a/setup.py
+++ b/setup.py
@@ -86,6 +86,7 @@
 extra_deps['databricks'] = [
     'mosaicml[databricks]>=0.17.1,<0.18',
     'databricks-sql-connector>=3, <4',
+    'databricks-connect==14.0.0'
 ]
 
 extra_deps['tensorboard'] = [