diff --git a/ci/script.sh b/ci/script.sh
index 5d5f1d2c6..67dcd4e89 100755
--- a/ci/script.sh
+++ b/ci/script.sh
@@ -3,7 +3,7 @@
 set -eux
 
 # Install rustup and the specified rust toolchain.
-curl https://sh.rustup.rs -sSf | sh -s -- --default-toolchain=$RUST_TOOLCHAIN -y
+curl https://sh.rustup.rs -sSf | sh -s -- --default-toolchain=$1 -y
 
 # Load cargo environment. Specifically, put cargo into PATH.
 source ~/.cargo/env
@@ -12,7 +12,7 @@ rustc --version
 rustup --version
 cargo --version
 
-case $TARGET in
+case $2 in
 	"native")
 		sudo apt-get -y update
 		sudo apt-get install -y cmake pkg-config libssl-dev