diff --git a/OneStage/yolo/deep_sort_yolov3/yolo3/model.py b/OneStage/yolo/deep_sort_yolov3/yolo3/model.py
index 44f1c68..a114954 100644
--- a/OneStage/yolo/deep_sort_yolov3/yolo3/model.py
+++ b/OneStage/yolo/deep_sort_yolov3/yolo3/model.py
@@ -7,7 +7,7 @@
 from keras import backend as K
 from keras.layers import Conv2D, Add, ZeroPadding2D, UpSampling2D, Concatenate
 from keras.layers.advanced_activations import LeakyReLU
-from keras.layers.normalization import BatchNormalization
+from tensorflow.keras.layers import BatchNormalization
 from keras.models import Model
 from keras.regularizers import l2