diff --git a/model/src/utils/load_model.py b/model/src/utils/load_model.py
index 332bea0..912db6e 100644
--- a/model/src/utils/load_model.py
+++ b/model/src/utils/load_model.py
@@ -5,7 +5,7 @@ from pathlib import Path
 
 import torch
 from minio import Minio
-from models import VisualCommunicationModel
+from src.models import VisualCommunicationModel
 
 from shared.data_store import get_model