diff --git a/model/src/main.py b/model/src/main.py
index aff4a23..6a8b572 100644
--- a/model/src/main.py
+++ b/model/src/main.py
@@ -8,14 +8,14 @@ from models import VisualCommunicationModel
 from tqdm import tqdm
 from utils import DEVICE, VCDADataset, load_model
 
-from shared.data_store import connect
+from shared.data_store import connect_minio
 from shared.utils import setup_logging
 
 if __name__ == '__main__':
     # setup logging
     setup_logging()
     # connect to minio
-    minio_client = connect()
+    minio_client = connect_minio()
     # instantiate model
     model: VisualCommunicationModel = load_model(client=minio_client)
     model.eval()