diff --git a/.azure-pipelines/multi-nodes-test.yml b/.azure-pipelines/multi-nodes-test.yml
index 7c9d35094..bcf517569 100644
--- a/.azure-pipelines/multi-nodes-test.yml
+++ b/.azure-pipelines/multi-nodes-test.yml
@@ -34,7 +34,7 @@ jobs:
     name: SshKeyFile
     displayName: Download key file
     inputs:
-      secureFile: ssh.key
+      secureFile: mscclpp-ssh.key
 
   - task: Bash@3
     name: InstallPackages
diff --git a/docker/base-dev-x.dockerfile b/docker/base-dev-x.dockerfile
index d7f2166f1..d6236cd02 100644
--- a/docker/base-dev-x.dockerfile
+++ b/docker/base-dev-x.dockerfile
@@ -1,4 +1,4 @@
-ARG BASE_IMAGE=ghcr.io/microsoft/mscclpp/mscclpp:base-cuda12.1
+ARG BASE_IMAGE
 FROM ${BASE_IMAGE}
 
 LABEL maintainer="MSCCL++"
diff --git a/docker/base-x.dockerfile b/docker/base-x.dockerfile
index bf29f718a..4be89c9d0 100644
--- a/docker/base-x.dockerfile
+++ b/docker/base-x.dockerfile
@@ -1,4 +1,4 @@
-ARG BASE_IMAGE=nvidia/cuda:12.1.1-devel-ubuntu20.04
+ARG BASE_IMAGE
 FROM ${BASE_IMAGE}
 
 LABEL maintainer="MSCCL++"