diff --git a/api/tests/unit_tests/core/app/test_llm_quota.py b/api/tests/unit_tests/core/app/test_llm_quota.py
index baf54851e9..13bdf76535 100644
--- a/api/tests/unit_tests/core/app/test_llm_quota.py
+++ b/api/tests/unit_tests/core/app/test_llm_quota.py
@@ -1,4 +1,4 @@
-from collections.abc import Iterator
+from collections.abc import Generator
 from contextlib import contextmanager
 from types import SimpleNamespace
 from unittest.mock import MagicMock, patch
@@ -26,7 +26,7 @@ from models.provider import Provider, ProviderType
 
 
 @contextmanager
-def _patched_credit_pool_session_factory(engine: Engine) -> Iterator[None]:
+def _patched_credit_pool_session_factory(engine: Engine) -> Generator[None, None, None]:
     session_maker = sessionmaker(bind=engine, expire_on_commit=False)
     with patch("services.credit_pool_service.session_factory.get_session_maker", return_value=session_maker):
         yield
diff --git a/api/tests/unit_tests/events/test_update_provider_when_message_created.py b/api/tests/unit_tests/events/test_update_provider_when_message_created.py
index ca21b0d205..4b2a6438f4 100644
--- a/api/tests/unit_tests/events/test_update_provider_when_message_created.py
+++ b/api/tests/unit_tests/events/test_update_provider_when_message_created.py
@@ -1,4 +1,4 @@
-from collections.abc import Iterator
+from collections.abc import Generator
 from contextlib import contextmanager
 from types import SimpleNamespace
 from unittest.mock import patch
@@ -16,7 +16,7 @@ from models.provider import ProviderType
 
 
 @contextmanager
-def _patched_credit_pool_session_factory(engine: Engine) -> Iterator[None]:
+def _patched_credit_pool_session_factory(engine: Engine) -> Generator[None, None, None]:
     session_maker = sessionmaker(bind=engine, expire_on_commit=False)
     with patch("services.credit_pool_service.session_factory.get_session_maker", return_value=session_maker):
         yield
diff --git a/api/tests/unit_tests/services/test_credit_pool_service.py b/api/tests/unit_tests/services/test_credit_pool_service.py
index 62e492c67d..6956dbbd6e 100644
--- a/api/tests/unit_tests/services/test_credit_pool_service.py
+++ b/api/tests/unit_tests/services/test_credit_pool_service.py
@@ -1,4 +1,4 @@
-from collections.abc import Iterator
+from collections.abc import Generator
 from contextlib import contextmanager
 from unittest.mock import patch
 from uuid import uuid4
@@ -34,7 +34,7 @@ def _create_engine_with_pool(*, quota_limit: int, quota_used: int) -> tuple[Engi
 
 
 @contextmanager
-def _patched_session_factory(engine: Engine) -> Iterator[None]:
+def _patched_session_factory(engine: Engine) -> Generator[None, None, None]:
     session_maker = sessionmaker(bind=engine, expire_on_commit=False)
     with patch("services.credit_pool_service.session_factory.get_session_maker", return_value=session_maker):
         yield