diff --git a/misc/image_upload_to_server.py b/misc/image_upload_to_server.py
index 6b0d57f..bef91af 100644
--- a/misc/image_upload_to_server.py
+++ b/misc/image_upload_to_server.py
@@ -6,7 +6,8 @@ from dotenv import load_dotenv
 from pymongo.errors import DuplicateKeyError
 
 from shared.data_store import connect_minio
-from shared.mongodb import VisualCommunication, connect_mongodb
+from shared.mongodb import connect_mongodb
+from shared.mongodb.classes import VisualCommunication
 from shared.utils import check_env, setup_logging
 from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 
diff --git a/other/transfer_images_mongo_minio.py b/other/transfer_images_mongo_minio.py
index 4d4d819..d308eed 100644
--- a/other/transfer_images_mongo_minio.py
+++ b/other/transfer_images_mongo_minio.py
@@ -11,7 +11,8 @@ from dotenv import load_dotenv
 from pymongo.collection import Collection
 
 from shared.data_store import connect_minio, put
-from shared.mongodb import VisualCommunication, connect_mongodb
+from shared.mongodb import connect_mongodb
+from shared.mongodb.classes import VisualCommunication
 from shared.utils import check_env, setup_logging
 from web_ui.src.main import NECESSARY_ENV_VAR_LIST