diff --git a/core/main.py b/core/main.py
index 12a6f36..cf3a3b3 100644
--- a/core/main.py
+++ b/core/main.py
@@ -1,11 +1,11 @@
 from core.api_server import APIServer
 from core.certificate_manager_mock import CertificateManagerMock
 from core.trace_id_handler import TraceIdHandler
-from shared.db_logger import DBLogger
+from shared.db_logger_mock import DBLoggerMock
 from shared.logger import Logger, TraceIdProvider
 
 logger = Logger(
-    TraceIdProvider(lambda: TraceIdHandler.get_current_trace_id()), DBLogger()
+    TraceIdProvider(lambda: TraceIdHandler.get_current_trace_id()), DBLoggerMock()
 )
 certificate_manager = CertificateManagerMock()
 api_server = APIServer(certificate_manager, logger, "0.0.1", 5000)
diff --git a/front/main.py b/front/main.py
index f6e1d74..144fe74 100644
--- a/front/main.py
+++ b/front/main.py
@@ -7,7 +7,7 @@ from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 from pydantic import BaseModel
 
-from api_client import APIClient
+from front.api_client import APIClient
 
 API_BASE_URL = "http://localhost:5000"
 app = FastAPI()