diff --git a/openai_patch/__init__.py b/openai_patch/__init__.py
index d88ddcfdbe9079f788cb6f2e58c0dcf10374e899..edfa2764d8aa75f254261155f58a0169e6a3c818 100644
--- a/openai_patch/__init__.py
+++ b/openai_patch/__init__.py
@@ -5,7 +5,7 @@ import logging
 from homeassistant.helpers import intent
 from homeassistant.exceptions import ServiceNotFound
 from homeassistant.components import conversation
-from homeassistant.components.openai_conversation import OpenAIConversationEntity
+from homeassistant.components.openai_conversation.conversation import OpenAIConversationEntity
 
 _LOGGER = logging.getLogger(__name__)