diff --git a/cuda_bindings/tests/nvml/test_compute_mode.py b/cuda_bindings/tests/nvml/test_compute_mode.py
index 6ef1818fe9..0e9a67861f 100644
--- a/cuda_bindings/tests/nvml/test_compute_mode.py
+++ b/cuda_bindings/tests/nvml/test_compute_mode.py
@@ -26,6 +26,6 @@ def test_compute_mode_supported_nonroot(all_devices):
             try:
                 nvml.device_set_compute_mode(device, cm)
             except nvml.NoPermissionError:
-                skip_reasons.add(f"nvmlDeviceSetComputeMode requires root for device {device}")
-                continue
+                pytest.skip("Insufficient permissions to set compute mode")
+            nvml.device_set_compute_mode(device, original_compute_mode)
             assert original_compute_mode == nvml.device_get_compute_mode(device), "Compute mode shouldn't have changed"