diff --git a/cognee/api/client.py b/cognee/api/client.py
index f35700427..b6ceea224 100644
--- a/cognee/api/client.py
+++ b/cognee/api/client.py
@@ -165,7 +165,7 @@ async def get_datasets(user: User = Depends(get_authenticated_user)):
 async def delete_dataset(dataset_id: str, user: User = Depends(get_authenticated_user)):
     from cognee.modules.data.methods import get_dataset, delete_dataset
 
-    dataset = get_dataset(user.id, dataset_id)
+    dataset = await get_dataset(user.id, dataset_id)
 
     if dataset is None:
         raise HTTPException(
diff --git a/cognee/modules/settings/get_settings.py b/cognee/modules/settings/get_settings.py
index 1571606a7..b15c0afad 100644
--- a/cognee/modules/settings/get_settings.py
+++ b/cognee/modules/settings/get_settings.py
@@ -16,7 +16,7 @@ class LLMConfig(BaseModel):
     api_key: str
     model: ConfigChoice
     provider: ConfigChoice
-    models: dict[ModelName, list[ConfigChoice]]
+    models: dict[str, list[ConfigChoice]]
     providers: list[ConfigChoice]
 
 class VectorDBConfig(BaseModel):