diff --git a/Backend/app/database/DatabaseTestingConnection.py b/Backend/app/database/DatabaseTestingConnection.py
index 2a64afc5..97a3b316 100644
--- a/Backend/app/database/DatabaseTestingConnection.py
+++ b/Backend/app/database/DatabaseTestingConnection.py
@@ -15,7 +15,7 @@ def _get_mongo_client(self):  # noqa: ANN202
         """Get mock Mongo client class
 
         Returns:
-            Any: the mock Mongo client class
+            MongoClientMock: the mock Mongo client class
         """
         enable_gridfs_integration()
         return MongoClientMock
diff --git a/Backend/app/database/database_schema.py b/Backend/app/database/database_schema.py
index 8e002ed3..b31878f4 100644
--- a/Backend/app/database/database_schema.py
+++ b/Backend/app/database/database_schema.py
@@ -31,6 +31,8 @@ class DatabaseCollection(StrEnum):
 
 
 def __is_connection__init__(func: Callable):  # noqa: ANN202
+    """Check database connection is initialized"""
+
     @wraps(func)
     def wrapper(*args, **kwargs):  # noqa: ANN202
         if BaseDatabaseConnection.connection is not None: