diff --git a/src/database/__init__.py b/src/database/__init__.py
index e790a76..3cf16e2 100644
--- a/src/database/__init__.py
+++ b/src/database/__init__.py
@@ -7,5 +7,7 @@ from .database import connect
 from .utils import (
     total_documents,
     total_annotated,
-    get_visual_communication
+    get_visual_communication,
+    upsert_annotations,
+    upsert_predictions,
 )