diff --git a/official/requirements.txt b/official/requirements.txt
index 3e30f54f30a..4b0c2143641 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