diff --git a/cli/jobs/single-step/tensorflow/mnist-distributed-horovod/job.yml b/cli/jobs/single-step/tensorflow/mnist-distributed-horovod/job.yml
index 856c9cfe160..4ce38729cea 100644
--- a/cli/jobs/single-step/tensorflow/mnist-distributed-horovod/job.yml
+++ b/cli/jobs/single-step/tensorflow/mnist-distributed-horovod/job.yml
@@ -5,7 +5,7 @@ command: >-
   --epochs ${{inputs.epochs}}
 inputs:
   epochs: 1
-environment: azureml:AzureML-tensorflow-2.12-cuda11@latest
+environment: azureml:AzureML-tensorflow-2.16-cuda11@latest
 compute: azureml:gpu-cluster
 resources:
   instance_count: 2