fixed imports

tests/total_annotated_test.py

@@ -6,7 +6,7 @@ from pathlib import Path
 from dotenv import load_dotenv
 
 from core.database import connect
-from core.database import total_documents
+from core.database import count_documents
 
 if __name__ == '__main__':
     # prepare env vars
@@ -17,7 +17,8 @@ if __name__ == '__main__':
     # connect to database
     collection, db, client = connect()
     # get visual communication
-    num_docs = total_documents(
+    num_docs = count_documents(
         collection=collection,
+        only_with_annotation=True,
     )
     print(f"number of annotated  documents in database: {num_docs}")