diff --git a/tests/api/helpers.py b/tests/api/helpers.py
index 8a78df0..3f962bc 100644
--- a/tests/api/helpers.py
+++ b/tests/api/helpers.py
@@ -1,7 +1,7 @@
 import contextlib
 import typing
 import uuid
-from collections.abc import Iterator
+from collections.abc import Generator
 
 import sqlalchemy as sa
 from httpx import AsyncClient
@@ -41,7 +41,7 @@ async def send(client: AsyncClient, chat_id: int, text: str, key: uuid.UUID | No
 
 
 @contextlib.contextmanager
-def count_statements(session: AsyncSession) -> Iterator[list[str]]:
+def count_statements(session: AsyncSession) -> Generator[list[str]]:
     """Collect every SQL statement sent on the test connection behind ``session`` while the block runs."""
     connection: typing.Final = session.bind
     assert isinstance(connection, AsyncConnection)
diff --git a/tests/conftest.py b/tests/conftest.py
index b667c39..dc087f5 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,3 +1,4 @@
+import collections.abc
 import contextlib
 import typing
 
@@ -40,7 +41,7 @@ async def di_container(app: litestar.Litestar) -> typing.AsyncIterator[modern_di
 
 
 @contextlib.asynccontextmanager
-async def rolled_back_db_session(di_container: modern_di.Container) -> typing.AsyncIterator[AsyncSession]:
+async def rolled_back_db_session(di_container: modern_di.Container) -> collections.abc.AsyncGenerator[AsyncSession]:
     engine = create_database_engine()
     connection = await engine.connect()
     transaction = await connection.begin()