diff --git a/tests-unit/server_test/test_cache_control.py b/tests/unit/test_cache_control.py
similarity index 99%
rename from tests-unit/server_test/test_cache_control.py
rename to tests/unit/test_cache_control.py
index 8de59125a..7e9b8a6f1 100644
--- a/tests-unit/server_test/test_cache_control.py
+++ b/tests/unit/test_cache_control.py
@@ -5,7 +5,7 @@ from aiohttp import web
 from aiohttp.test_utils import make_mocked_request
 from typing import Dict, Any
 
-from middleware.cache_middleware import cache_control, ONE_HOUR, ONE_DAY, IMG_EXTENSIONS
+from comfy.middleware.cache_middleware import cache_control, ONE_HOUR, ONE_DAY, IMG_EXTENSIONS
 
 pytestmark = pytest.mark.asyncio  # Apply asyncio mark to all tests