diff --git a/custom_components/ha_text_ai/__init__.py b/custom_components/ha_text_ai/__init__.py
index acfa351..9e24069 100644
--- a/custom_components/ha_text_ai/__init__.py
+++ b/custom_components/ha_text_ai/__init__.py
@@ -4,7 +4,7 @@ from __future__ import annotations
 import logging
 import os
 import shutil
-import voluptuous as vol  
+import voluptuous as vol
 from typing import Any, Dict, Optional
 from datetime import datetime
 
diff --git a/custom_components/ha_text_ai/coordinator.py b/custom_components/ha_text_ai/coordinator.py
index ab33873..1c02474 100644
--- a/custom_components/ha_text_ai/coordinator.py
+++ b/custom_components/ha_text_ai/coordinator.py
@@ -14,7 +14,6 @@ import voluptuous as vol
 from homeassistant import config_entries
 from homeassistant.helpers import aiohttp_client
 from homeassistant.exceptions import HomeAssistantError
-from homeassistant.exceptions import HomeAssistantError, ConfigUpdateFailed
 from openai import AsyncOpenAI, APIError, AuthenticationError, RateLimitError
 from anthropic import AsyncAnthropic
 from homeassistant.core import HomeAssistant
@@ -560,7 +559,7 @@ class HATextAICoordinator(DataUpdateCoordinator):
         except Exception as e:
             self._last_error = str(e)
             _LOGGER.error("Error updating data: %s", str(e))
-            raise ConfigUpdateFailed(f"Error updating data: {str(e)}")
+            raise HomeAssistantError(f"Error updating data: {str(e)}")
 
     async def __aenter__(self):
         """Async enter."""