diff --git a/src/database/__init__.py b/src/database/__init__.py
index 3cf16e2..0990ebc 100644
--- a/src/database/__init__.py
+++ b/src/database/__init__.py
@@ -1,13 +1,12 @@
-from .classes import (
-    ModelOutputs,
-    VisualCommunication,
-    NoDocumentFoundException
-)
+from __future__ import annotations
+
+from .classes import ModelOutputs
+from .classes import NoDocumentFoundException
+from .classes import VisualCommunication
 from .database import connect
-from .utils import (
-    total_documents,
-    total_annotated,
-    get_visual_communication,
-    upsert_annotations,
-    upsert_predictions,
-)
+from .utils import get_visual_communication
+from .utils import total_annotated
+from .utils import total_documents
+from .utils import upsert_annotations
+from .utils import upsert_predictions
+from .utils import upsert_visual_communication