diff --git a/tensorflow/tools/dist_test/Dockerfile b/tensorflow/tools/dist_test/Dockerfile
index 2a7605bbc960f1..9f21daee70752c 100644
--- a/tensorflow/tools/dist_test/Dockerfile
+++ b/tensorflow/tools/dist_test/Dockerfile
@@ -18,7 +18,7 @@
 #
 # See ./remote_test.sh for usage example.
 
-FROM ubuntu:16.04
+FROM ubuntu:xenial-20210114
 
 LABEL maintainer="Shanqing Cai <cais@google.com>"