diff --git a/model/src/models/visual_communication.py b/model/src/models/visual_communication.py
index 0a841d2..256203b 100644
--- a/model/src/models/visual_communication.py
+++ b/model/src/models/visual_communication.py
@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from torch import nn
 
-from shared.mongodb.classes import ModelData
+from shared.mongodb.src.classes import ModelData
 
 from .angle import AngleTail
 from .contact import ContactTail