diff --git a/requirements.txt b/requirements.txt
index 81af83f7b..8d1007a3b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 databricks-sql-connector>=2.9.3, <3.0.0
-dbt-spark==1.7.0b1
+dbt-spark==1.7.0b2
 databricks-sdk==0.9.0
 keyring>=23.13.0
diff --git a/setup.py b/setup.py
index 6abc9dcf6..94097418e 100644
--- a/setup.py
+++ b/setup.py
@@ -54,7 +54,8 @@ def _get_plugin_version():
     packages=find_namespace_packages(include=["dbt", "dbt.*"]),
     include_package_data=True,
     install_requires=[
-        "dbt-spark~=1.7.0b1",
+        # "dbt-spark~=1.7.0b1",
+        "dbt-spark @ git+ssh://git@github.com/dbt-labs/dbt-spark.git@main",
         "databricks-sql-connector>=2.9.3, <3.0.0",
         "databricks-sdk>=0.9.0",
         "keyring>=23.13.0",
diff --git a/tests/profiles.py b/tests/profiles.py
index 73c660ff4..a96dc90ae 100644
--- a/tests/profiles.py
+++ b/tests/profiles.py
@@ -21,6 +21,7 @@ def _build_databricks_cluster_target(
     profile: Dict[str, Any] = {
         "type": "databricks",
         "host": os.getenv("DBT_DATABRICKS_HOST_NAME"),
+        "database": os.getenv("DBT_DATABRICKS_DB_NAME"),
         "http_path": http_path,
         "token": os.getenv("DBT_DATABRICKS_TOKEN"),
         "client_id": os.getenv("DBT_DATABRICKS_CLIENT_ID"),