diff --git a/tests/image_download_test.py b/tests/image_download_test.py
index 7020013..f84cd25 100644
--- a/tests/image_download_test.py
+++ b/tests/image_download_test.py
@@ -5,7 +5,7 @@ from pathlib import Path
 
 from dotenv import load_dotenv
 
-from shared.database import connect
+from shared.database import connect_mongodb
 from shared.database.classes import VisualCommunication
 
 if __name__ == '__main__':
@@ -15,7 +15,7 @@ if __name__ == '__main__':
     load_dotenv(env_path)
     os.environ['MONGO_HOST'] = 'localhost'
     # connect to database
-    collection, db, client = connect()
+    collection, db, client = connect_mongodb()
     print(client.server_info())
     # download images
     data = None