diff --git a/api/tests/unit_tests/core/test_model_manager.py b/api/tests/unit_tests/core/test_model_manager.py
index 1f2f258fe6..96176bb6c4 100644
--- a/api/tests/unit_tests/core/test_model_manager.py
+++ b/api/tests/unit_tests/core/test_model_manager.py
@@ -2,7 +2,7 @@ from unittest.mock import MagicMock
 
 import pytest
 import redis
-from api.extensions.ext_redis import RedisClientWrapper
+from extensions.ext_redis import RedisClientWrapper
 
 from core.entities.provider_entities import ModelLoadBalancingConfiguration
 from core.model_manager import LBModelManager