diff --git a/OneStage/yolo/deep_sort_yolov3/deep_sort/linear_assignment.py b/OneStage/yolo/deep_sort_yolov3/deep_sort/linear_assignment.py
index 178456c..96a8794 100644
--- a/OneStage/yolo/deep_sort_yolov3/deep_sort/linear_assignment.py
+++ b/OneStage/yolo/deep_sort_yolov3/deep_sort/linear_assignment.py
@@ -1,7 +1,7 @@
 # vim: expandtab:ts=4:sw=4
 from __future__ import absolute_import
 import numpy as np
-from sklearn.utils.linear_assignment_ import linear_assignment
+from scipy.optimize import linear_sum_assignment as linear_assignment
 from . import kalman_filter