diff --git a/requirements-gpu.txt b/requirements-gpu.txt
index 49db6bb..1ff1d7e 100644
--- a/requirements-gpu.txt
+++ b/requirements-gpu.txt
@@ -7,4 +7,4 @@ tensorflow-gpu==1.15.4
 tensorflow-estimator==1.15.1
 tqdm==4.31.1
 requests==2.22.0
-ujson==2.0.3
\ No newline at end of file
+ujson==5.4.0
\ No newline at end of file
diff --git a/requirements-tpu.txt b/requirements-tpu.txt
index 1955b8f..b1d5e65 100644
--- a/requirements-tpu.txt
+++ b/requirements-tpu.txt
@@ -7,4 +7,4 @@ tensorflow==1.15.4
 tensorflow-estimator==1.15.1
 tqdm==4.31.1
 requests==2.22.0
-ujson==2.0.3
\ No newline at end of file
+ujson==5.4.0
\ No newline at end of file