diff --git a/_tests/ai/test_gemini_client.py b/_tests/ai/test_gemini_client.py
index de7797f..24ca928 100644
--- a/_tests/ai/test_gemini_client.py
+++ b/_tests/ai/test_gemini_client.py
@@ -1,7 +1,7 @@
 import pytest
 from unittest.mock import patch, MagicMock, AsyncMock
 
-from ml_models.gemini import (
+from app.core.ml_models.gemini import (
     GeminiClient,
     GeminiModel,
     AuthenticationError,