diff --git a/tensorflow/tools/ci_build/Dockerfile.micro b/tensorflow/tools/ci_build/Dockerfile.micro
index a86b4c828b5..cf2d7155f94 100644
--- a/tensorflow/tools/ci_build/Dockerfile.micro
+++ b/tensorflow/tools/ci_build/Dockerfile.micro
@@ -1,11 +1,11 @@
 # Use a prebuilt Python image instead of base Ubuntu to speed up the build process,
 # since it has all the build dependencies we need for Micro and downloads much faster
 # than the install process.
-FROM python:3.5-stretch
+FROM python:3.9.0-buster
 
 LABEL maintainer="Pete Warden <petewarden@google.com>"
 
 RUN apt-get update && apt-get install -y zip xxd
-RUN pip install six 
+RUN pip install six
 # Install Renode test dependencies
 RUN pip install pyyaml requests psutil robotframework==3.1