diff --git a/scripts/build.sh b/scripts/build.sh
index e2734ad..9a1c633 100755
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -3,8 +3,8 @@
 set -e
 
 CUDA_VERSION=$(nvcc --version | grep release | sed -E 's/.*release ([0-9]+\.[0-9]+).*/\1/')
-CUDA_MAJOR=${CUDA_VERSION_RAW%.*}
-CUDA_MINOR=${CUDA_VERSION_RAW#*.}
+CUDA_MAJOR=${CUDA_VERSION%.*}
+CUDA_MINOR=${CUDA_VERSION#*.}
 
 CUDA_ARCH_LIST="8.0;8.6"