diff --git a/api_integration.py b/api_integration.py
index 9fdf93e..a67b984 100644
--- a/api_integration.py
+++ b/api_integration.py
@@ -9,8 +9,8 @@ import json
 sys.path.append(os.path.join(os.path.dirname(__file__), 'api_service'))
 
 # Import the API client and models
-from discord_client import ApiClient
-from api_models import Conversation, UserSettings, Message
+from api_service.discord_client import ApiClient
+from api_service.api_models import Conversation, UserSettings, Message
 
 # API client instance
 api_client = None
diff --git a/api_service/api_server.py b/api_service/api_server.py
index 94ce150..768297d 100644
--- a/api_service/api_server.py
+++ b/api_service/api_server.py
@@ -277,7 +277,7 @@ async def send_discord_message_via_api(channel_id: int, content: str, timeout: f
 # Import dependencies after defining settings and constants
 # Use absolute imports to avoid issues when running the server directly
 from api_service import dependencies # type: ignore
-from api_models import (
+from api_service.api_models import (
     Conversation,
     UserSettings,
     GetConversationsResponse,
diff --git a/api_service/discord_client.py b/api_service/discord_client.py
index 17862d0..d0b0359 100644
--- a/api_service/discord_client.py
+++ b/api_service/discord_client.py
@@ -2,7 +2,7 @@ import aiohttp
 import json
 import datetime
 from typing import Dict, List, Optional, Any, Union
-from api_models import Conversation, UserSettings, Message
+from api_service.api_models import Conversation, UserSettings, Message
 
 class ApiClient:
     def __init__(self, api_url: str, token: Optional[str] = None):