diff --git a/OneStage/yolo/deep_sort_yolov3/convert.py b/OneStage/yolo/deep_sort_yolov3/convert.py
index 8723e66..677f209 100644
--- a/OneStage/yolo/deep_sort_yolov3/convert.py
+++ b/OneStage/yolo/deep_sort_yolov3/convert.py
@@ -15,7 +15,7 @@
 from keras.layers import (Conv2D, Input, ZeroPadding2D, Add,
                           UpSampling2D, MaxPooling2D, 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
 from keras.utils.vis_utils import plot_model as plot