diff --git a/custom_components/ha_text_ai/config_flow.py b/custom_components/ha_text_ai/config_flow.py
index f46bfda..0146e49 100644
--- a/custom_components/ha_text_ai/config_flow.py
+++ b/custom_components/ha_text_ai/config_flow.py
@@ -6,8 +6,7 @@ from typing import Any
 
 import voluptuous as vol  
 from openai import AsyncOpenAI  
-from openai import OpenAIError  # Updated import  
-from openai.exceptions import AuthenticationError, APIConnectionError  
+from openai import OpenAIError, AuthenticationError, APIConnectionError  
 
 from homeassistant import config_entries  
 from homeassistant.core import HomeAssistant, callback  
@@ -25,16 +24,10 @@ from .const import (
 
 _LOGGER = logging.getLogger(__name__)  
 
-@config_entries.HANDLERS.register(DOMAIN)  
 class TextAIConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):  
     """Handle a config flow for HA Text AI Integration."""  
 
     VERSION = 1  
-    CONNECTION_CLASS = config_entries.CONN_CLASS_CLOUD_POLL  
-
-    def __init__(self) -> None:  
-        """Initialize the config flow."""  
-        super().__init__()  
 
     async def async_step_user(  
         self, user_input: dict[str, Any] | None = None  
@@ -86,8 +79,8 @@ class TextAIConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
             raise ApiKeyError from err  
         except APIConnectionError as err:  
             raise ApiConnectionError from err  
-        except OpenAIError as err:  # Changed from APIError  
-            if getattr(err, 'status_code', None) == 401:  # Safe status_code check  
+        except OpenAIError as err:  
+            if getattr(err, 'status_code', None) == 401:  
                 raise ApiKeyError from err  
             raise ApiConnectionError from err