diff --git a/release/long_running_distributed_tests/cluster.yaml b/release/long_running_distributed_tests/cluster.yaml
index 4710a47fc..077d442d6 100644
--- a/release/long_running_distributed_tests/cluster.yaml
+++ b/release/long_running_distributed_tests/cluster.yaml
@@ -6,7 +6,7 @@ max_workers: 3
 idle_timeout_minutes: 15
 
 docker:
-    image: anyscale/ray-ml:latest-gpu
+    image: rayproject/ray-ml:latest-gpu
     container_name: ray_container
     pull_before_run: True
 
@@ -21,14 +21,16 @@ auth:
 
 head_node:
     InstanceType: g3.8xlarge
+    ImageId: ami-0828a1066dc750737
 
 worker_nodes:
   InstanceType: g3.8xlarge
+  ImageId: ami-0828a1066dc750737
   InstanceMarketOptions:
     MarketType: spot
 
 setup_commands:
-  - apt-get install -y libglib2.0-0 libcudnn7=7.6.5.32-1+cuda10.1
+  - sudo apt-get install -y libglib2.0-0 libcudnn7=7.6.5.32-1+cuda10.1
   - pip install -U https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-2.0.0.dev0-cp37-cp37m-manylinux2014_x86_64.whl
 
 # Command to start ray on the head node. You don't need to change this.
diff --git a/release/long_running_distributed_tests/run.sh b/release/long_running_distributed_tests/run.sh
index d0fa4a6c4..4e0b34e68 100755
--- a/release/long_running_distributed_tests/run.sh
+++ b/release/long_running_distributed_tests/run.sh
@@ -43,6 +43,7 @@ echo "branch: $ray_branch"
 echo "workload: $workload"
 
 wheel="https://s3-us-west-2.amazonaws.com/ray-wheels/$ray_branch/$commit/ray-$ray_version-cp37-cp37m-manylinux2014_x86_64.whl"
+wheel="https://ray-wheels.s3-us-west-2.amazonaws.com/releases/1.2.0/b4d87b8fc5ff03b868adba35b37da32b7be9f71f/ray-1.2.0-cp37-cp37m-manylinux2014_x86_64.whl"
 
 conda uninstall -y terminado || true
 pip install -U pip
diff --git a/release/stress_tests/autoscaler-cluster.yaml b/release/stress_tests/autoscaler-cluster.yaml
index 9c17d303e..6e73f24cb 100644
--- a/release/stress_tests/autoscaler-cluster.yaml
+++ b/release/stress_tests/autoscaler-cluster.yaml
@@ -91,7 +91,7 @@ setup_commands:
     - pip install terminado
     - pip install boto3==1.4.8 cython==0.29.0
     # - cd ray/python; git checkout master; git pull; pip install -e . --verbose
-    - pip install -U pip install https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-2.0.0.dev0-cp38-cp38-manylinux2014_x86_64.whl
+    - pip install -U pip install https://ray-wheels.s3-us-west-2.amazonaws.com/releases/1.2.0/b4d87b8fc5ff03b868adba35b37da32b7be9f71f/ray-1.2.0-cp37-cp37m-manylinux2014_x86_64.whl
 
 # Custom commands that will be run on the head node after common setup.
 head_setup_commands: []