diff --git a/components/configs/kustomized/nvidia-gpu-verification/base/deployment.yaml b/components/configs/kustomized/nvidia-gpu-verification/base/deployment.yaml
index e99d57ac..fe97fbc1 100644
--- a/components/configs/kustomized/nvidia-gpu-verification/base/deployment.yaml
+++ b/components/configs/kustomized/nvidia-gpu-verification/base/deployment.yaml
@@ -15,7 +15,7 @@ spec:
         app: nvidia-vector-add
     spec:
 #      tolerations:
-#        - key: nvidia.com/gpu
+#        - key: nvidia-gpu-only
 #          operator: Exists
 #          effect: NoSchedule
       containers:
diff --git a/scripts/library/ocp.sh b/scripts/library/ocp.sh
index e826ca25..4767e04d 100644
--- a/scripts/library/ocp.sh
+++ b/scripts/library/ocp.sh
@@ -136,9 +136,9 @@ ocp_aws_create_gpu_machineset(){
     --type=merge --patch '{"spec":{"template":{"spec":{"metadata":{"labels":{"node-role.kubernetes.io/gpu":""}}}}}}'
 
   # taint nodes for gpu-only workloads
-  # oc -n openshift-machine-api \
-  #   patch "${MACHINE_SET_GPU}" \
-  #   --type=merge --patch '{"spec":{"template":{"spec":{"taints":[{"key":nvidia.com/gpu","value":"","effect":"NoSchedule"}]}}}}'
+  oc -n openshift-machine-api \
+    patch "${MACHINE_SET_GPU}" \
+    --type=merge --patch '{"spec":{"template":{"spec":{"taints":[{"key":nvidia-gpu-only","value":"","effect":"NoSchedule"}]}}}}'
   
   # should use the default profile
   # oc -n openshift-machine-api \