diff --git a/model/src/min_io/__init__.py b/model/src/data_store/__init__.py
similarity index 100%
rename from model/src/min_io/__init__.py
rename to model/src/data_store/__init__.py
diff --git a/model/src/min_io/connect.py b/model/src/data_store/connect.py
similarity index 100%
rename from model/src/min_io/connect.py
rename to model/src/data_store/connect.py
diff --git a/model/src/min_io/get.py b/model/src/data_store/get.py
similarity index 100%
rename from model/src/min_io/get.py
rename to model/src/data_store/get.py
diff --git a/model/src/min_io/put.py b/model/src/data_store/put.py
similarity index 100%
rename from model/src/min_io/put.py
rename to model/src/data_store/put.py
diff --git a/model/src/main.py b/model/src/main.py
index b80361c..c8fc5d5 100644
--- a/model/src/main.py
+++ b/model/src/main.py
@@ -5,8 +5,8 @@ from hashlib import md5
 from io import BytesIO
 
 import torch
-from min_io import connect
-from min_io import put
+from data_store import connect
+from data_store import put
 from models import VisualCommunicationModel
 from torchinfo import summary
 from utils import check_env