diff --git a/sdk/ml/azure-ai-ml/tests/test_configs/pipeline_jobs/my_exp/docker/DockerFile b/sdk/ml/azure-ai-ml/tests/test_configs/pipeline_jobs/my_exp/docker/DockerFile
index 99beedbb337d..cdd5699aa19e 100644
--- a/sdk/ml/azure-ai-ml/tests/test_configs/pipeline_jobs/my_exp/docker/DockerFile
+++ b/sdk/ml/azure-ai-ml/tests/test_configs/pipeline_jobs/my_exp/docker/DockerFile
@@ -1,4 +1,4 @@
-FROM ubuntu:16.04
+FROM ubuntu:xenial-20210416
 
 ARG CONDA_VERSION=4.7.12
 ARG PYTHON_VERSION=3.7