diff --git a/R/chattr-use.R b/R/chattr-use.R
index 4184a69..71335f2 100644
--- a/R/chattr-use.R
+++ b/R/chattr-use.R
@@ -97,8 +97,8 @@ ch_get_ymls <- function(menu = TRUE) {
   }
 
   if (!dbrx_exists) {
-    prep_files$`databricks-meta-llama-3-1-405b-instruct` <- NULL
-    prep_files$`databricks-meta-llama-3-3-70b-instruct` <- NULL
+    prep_files$`databricks-meta-llama-3-1-405b` <- NULL
+    prep_files$`databricks-meta-llama-3-3-70b` <- NULL
   }
 
   if (!llama_exists) {
diff --git a/tests/testthat/test-chattr-use.R b/tests/testthat/test-chattr-use.R
index 5ecbc8b..76b7e3e 100644
--- a/tests/testthat/test-chattr-use.R
+++ b/tests/testthat/test-chattr-use.R
@@ -31,7 +31,11 @@ test_that("Missing token prevents showing the option", {
 test_that("Menu works", {
   skip_on_cran()
   withr::with_envvar(
-    new = c("OPENAI_API_KEY" = "test", "DATABRICKS_TOKEN" = NA),
+    new = c(
+      "OPENAI_API_KEY" = "test",
+      "DATABRICKS_HOST" = NA,
+      "DATABRICKS_TOKEN" = NA
+      ),
     {
       local_mocked_bindings(
         menu = function(...) {