diff --git a/requirements.txt b/requirements.txt
index 48bd83d..724e437 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,5 +9,5 @@ ffmpeg-python
 wget
 requests
 flask>=1.1.1
-werkzeug==0.16.1 # It fixes the error: from werkzeug import FileStorage cannot import FileStorage.
+werkzeug>=1.0.1
 cached_property
diff --git a/src/mot/serving/inference.py b/src/mot/serving/inference.py
index 4da8670..2c3dfdb 100644
--- a/src/mot/serving/inference.py
+++ b/src/mot/serving/inference.py
@@ -8,7 +8,7 @@
 import cv2
 import numpy as np
 from tqdm import tqdm
-from werkzeug import FileStorage
+from werkzeug.datastructures import FileStorage
 
 from mot.object_detection.query_server import \
     localizer_tensorflow_serving_inference
diff --git a/src/mot/serving/utils.py b/src/mot/serving/utils.py
index 14b38a8..c9aaf3d 100644
--- a/src/mot/serving/utils.py
+++ b/src/mot/serving/utils.py
@@ -1,6 +1,6 @@
 import os
 
-from werkzeug import FileStorage
+from werkzeug.datastructures import FileStorage
 from werkzeug.utils import secure_filename