diff --git a/install.sh b/install.sh
index fdd003a9..b376361d 100644
--- a/install.sh
+++ b/install.sh
@@ -65,7 +65,7 @@ while [[ $# -gt 0 ]]; do
         esac
         shift 2
         ;;
-    --cuda)
+    --device)
         case "$2" in
         CU124)
             CUDA_VERSION=124