diff --git a/requirements.txt b/requirements.txt
index 5ce99d65665..bcffd54124a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -15,3 +15,4 @@ distro==1.5.0
 tensorflow>=2.3.0,<2.4.0
 tensorflow-serving-api==2.3.0
 apache-beam==2.26.0
+google-apitools==0.5.31
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 1d2d3204454..ae0dcbb908c 100644
--- a/setup.py
+++ b/setup.py
@@ -68,6 +68,7 @@
         "tensorflow>=2.3.0,<2.4.0",
         "tensorflow-serving-api==2.3.0",
         "apache-beam==2.26.0",  # temporary for dataflow runner
+        "google-apitools==0.5.31",  # temporary for dataflow runner
     ],
     entry_points="""
     [console_scripts]