diff --git a/cli/jobs/single-step/tensorflow/mnist/job.yml b/cli/jobs/single-step/tensorflow/mnist/job.yml
index fa3df23574a..76726421acf 100644
--- a/cli/jobs/single-step/tensorflow/mnist/job.yml
+++ b/cli/jobs/single-step/tensorflow/mnist/job.yml
@@ -2,7 +2,7 @@ $schema: https://azuremlschemas.azureedge.net/latest/commandJob.schema.json
 code: src
 command: >-
   python train.py
-environment: azureml:AzureML-tensorflow-2.12-cuda11@latest
+environment: azureml:AzureML-tensorflow-2.16-cuda11@latest
 compute: azureml:gpu-cluster
 display_name: tensorflow-mnist-example
 experiment_name: tensorflow-mnist-example