diff --git a/tensorflow/tools/dist_test/Dockerfile b/tensorflow/tools/dist_test/Dockerfile
index 2a7605bbc960f1..c6be14ebf0a027 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-20210416
 
 LABEL maintainer="Shanqing Cai <cais@google.com>"