diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py
index 87a72a98..11c333ec 100644
--- a/lightrag/api/lightrag_server.py
+++ b/lightrag/api/lightrag_server.py
@@ -719,14 +719,14 @@ def create_app(args):
 
     # Create working directory if it doesn't exist
     Path(args.working_dir).mkdir(parents=True, exist_ok=True)
-    if args.llm_binding_host == "lollms" or args.embedding_binding == "lollms":
+    if args.llm_binding == "lollms" or args.embedding_binding == "lollms":
         from lightrag.llm.lollms import lollms_model_complete, lollms_embed
-    if args.llm_binding_host == "ollama" or args.embedding_binding == "ollama":
+    if args.llm_binding == "ollama" or args.embedding_binding == "ollama":
         from lightrag.llm.ollama import ollama_model_complete, ollama_embed
-    if args.llm_binding_host == "openai" or args.embedding_binding == "openai":
+    if args.llm_binding == "openai" or args.embedding_binding == "openai":
         from lightrag.llm.openai import openai_complete_if_cache, openai_embed
     if (
-        args.llm_binding_host == "azure_openai"
+        args.llm_binding == "azure_openai"
         or args.embedding_binding == "azure_openai"
     ):
         from lightrag.llm.azure_openai import (