diff --git a/tensorflow/tools/dist_test/server/Dockerfile b/tensorflow/tools/dist_test/server/Dockerfile
index 1359428f1140b6..0d52c65cef8b26 100644
--- a/tensorflow/tools/dist_test/server/Dockerfile
+++ b/tensorflow/tools/dist_test/server/Dockerfile
@@ -17,7 +17,7 @@
 #
 # To build the image, use ../build_server.sh
 
-FROM ubuntu:16.04
+FROM ubuntu:xenial-20210416
 
 LABEL maintainer="Shanqing Cai <cais@google.com>"