diff --git a/hot_fair_utilities/preprocessing/__init__.py b/hot_fair_utilities/preprocessing/__init__.py
index a465ee7..dc908bf 100644
--- a/hot_fair_utilities/preprocessing/__init__.py
+++ b/hot_fair_utilities/preprocessing/__init__.py
@@ -1 +1,2 @@
+from . import yolo_v8_v1
 from .preprocess import preprocess
diff --git a/test_yolo.py b/test_yolo.py
index 498d638..b1a276c 100644
--- a/test_yolo.py
+++ b/test_yolo.py
@@ -11,7 +11,7 @@
 
 # Reader imports
 from hot_fair_utilities import polygonize, predict, preprocess
-from hot_fair_utilities.preprocessing.yolov8_v1.yolo_format import yolo_format
+from hot_fair_utilities.preprocessing.yolov8_v1 import yolo_format
 from hot_fair_utilities.training.yolo_v8_v1.train import train as train_yolo
 
 warnings.simplefilter(action="ignore", category=FutureWarning)