diff --git a/llm_venice.py b/llm_venice.py
index 34d4746..60aed5e 100644
--- a/llm_venice.py
+++ b/llm_venice.py
@@ -41,12 +41,7 @@ def refresh():
         )
         response.raise_for_status()
         models = response.json()["data"]
-        text_models = [
-            model["id"]
-            for model in models
-            if model.get("type", {})
-            == "text"
-        ]
+        text_models = [model["id"] for model in models if model.get("type") == "text"]
         if not text_models:
             raise click.ClickException("No text generation models found")
         path = llm.user_dir() / "llm-venice.json"