diff --git a/model/src/utils/__init__.py b/model/src/utils/__init__.py
index aa29844..54c4ddf 100644
--- a/model/src/utils/__init__.py
+++ b/model/src/utils/__init__.py
@@ -1 +1,2 @@
-from .load_model import load_model
+from .load_model import DEVICE, load_model
+from .vcda_dataset import VCDADataset