diff --git a/simplemind/models.py b/simplemind/models.py
index e548435..95df38c 100644
--- a/simplemind/models.py
+++ b/simplemind/models.py
@@ -5,7 +5,7 @@ from typing import Any, Dict, List, Literal, Optional
 
 from pydantic import BaseModel, Field
 
-from .providers import find_provider
+from .utils import find_provider
 
 MESSAGE_ROLE = Literal["system", "user", "assistant"]
 
diff --git a/simplemind/providers/gemini.py b/simplemind/providers/gemini.py
index 1e7a932..b2f145d 100644
--- a/simplemind/providers/gemini.py
+++ b/simplemind/providers/gemini.py
@@ -5,7 +5,6 @@ import google.generativeai as genai
 import instructor
 from pydantic import BaseModel
 
-from ..models import Conversation, Message
 from ..settings import settings
 from ._base import BaseProvider
 
@@ -38,6 +37,7 @@ class Gemini(BaseProvider):
 
     def send_conversation(self, conversation: "Conversation") -> "Message":
         """Send a conversation to the Gemini API."""
+        from ..models import Message
 
         messages = [
             {