diff --git a/recognization/code/tracking.py b/recognization/code/tracking.py
index 01699ba..a77bc61 100644
--- a/recognization/code/tracking.py
+++ b/recognization/code/tracking.py
@@ -28,9 +28,8 @@ current_dir = os.path.dirname(os.path.abspath(__file__))
 if current_dir not in sys.path:
     sys.path.insert(0, current_dir)
 
-from ultralytics.trackers.basetrack import BaseTrack, STrack
-from ultralytics.trackers.byte_tracker import BYTETracker
-from ultralytics.utils import IterableSimpleNamespace, yaml_load
+from ultralytics.trackers.byte_tracker import BYTETracker, STrack
+from ultralytics.utils import YAML, IterableSimpleNamespace
 
 _BYTETRACK_YAML = os.path.join(
     os.path.dirname(os.path.abspath(__import__("ultralytics").__file__)),
@@ -50,7 +49,7 @@ DEFAULT_TRACKER_CONFIG = {
 
 def _default_args() -> IterableSimpleNamespace:
     try:
-        return IterableSimpleNamespace(**yaml_load(_BYTETRACK_YAML))
+        return IterableSimpleNamespace(**YAML.load(_BYTETRACK_YAML))
     except Exception:
         return IterableSimpleNamespace(**DEFAULT_TRACKER_CONFIG)