diff --git a/official/requirements.txt b/official/requirements.txt
index 3e30f54f30ac21ad64e9a1c66ad5f858d23c8eba..4b0c2143641742b17a2b4bcd41131255d4a45b03 100644
--- a/official/requirements.txt
+++ b/official/requirements.txt
@@ -11,7 +11,8 @@ py-cpuinfo>=3.3.0
 scipy>=0.19.1
 tensorflow-hub>=0.6.0
 tensorflow-model-optimization>=0.2.1
-tensorflow_datasets
+tensorflow-datasets
+tensorflow-addons
 dataclasses
 gin-config
 typing