diff --git a/test/sample-test/Dockerfile b/test/sample-test/Dockerfile
index 0b789b99e52..40dc51edb7b 100644
--- a/test/sample-test/Dockerfile
+++ b/test/sample-test/Dockerfile
@@ -1,11 +1,10 @@
 # This image has the script to kick off the ML pipeline sample e2e test,
 
-# TODO: use google/cloud-sdk:272.0.0, currently some python dependencies break if we use the new version
-FROM google/cloud-sdk:236.0.0
+FROM google/cloud-sdk:272.0.0
 
-RUN apt-get update -y && \
-    apt-get install --no-install-recommends -y -q default-jdk python3-setuptools python3-dev && \
-    apt-get install --no-install-recommends -y -q libssl-dev libffi-dev wget ssh
+RUN apt-get update -y
+RUN apt-get install --no-install-recommends -y -q libssl-dev libffi-dev wget ssh
+RUN apt-get install --no-install-recommends -y -q default-jre default-jdk python3-setuptools python3-dev gcc
 
 RUN wget https://bootstrap.pypa.io/get-pip.py && python3 get-pip.py