diff --git a/tests/test_model_regression.py b/tests/test_model_regression.py
index 114d86d..9875adb 100644
--- a/tests/test_model_regression.py
+++ b/tests/test_model_regression.py
@@ -4,7 +4,7 @@
 import pytest
 
 # bytetracker
-from bytetracker.byte_tracker import BYTETracker
+from bytetracker import BYTETracker
 
 # static paths
 OUTPUT_FOLDER = Path("tests/output")