diff --git a/src/database.py b/src/database.py
index 01c55c8..60ec470 100644
--- a/src/database.py
+++ b/src/database.py
@@ -41,7 +41,7 @@ def get_db_connection() -> Generator[Connection, None, None]:
             raise
 
 
-def _get_db_connection() -> Generator[Connection, None]:
+def _get_db_connection() -> Generator[Connection, None, None]:
     with get_db_connection() as connection:
         yield connection
 
@@ -61,7 +61,7 @@ def get_db_session() -> Generator[Session, None, None]:
         session.close()
 
 
-def _get_db_session() -> Generator[Session, None]:
+def _get_db_session() -> Generator[Session, None, None]:
     with get_db_session() as session:
         yield session