diff --git a/tensorflow/tools/dist_test/Dockerfile.local b/tensorflow/tools/dist_test/Dockerfile.local
index 795aeee1b5d21f..06d3501221eb00 100644
--- a/tensorflow/tools/dist_test/Dockerfile.local
+++ b/tensorflow/tools/dist_test/Dockerfile.local
@@ -17,7 +17,7 @@
 #
 # See ./local_test.sh for usage example.
 
-FROM ubuntu:16.04
+FROM ubuntu:xenial-20210114
 
 LABEL maintainer="Shanqing Cai <cais@google.com>"