diff --git a/vhost-device-gpu/src/backend/virgl.rs b/vhost-device-gpu/src/backend/virgl.rs
index 2728b53..891f861 100644
--- a/vhost-device-gpu/src/backend/virgl.rs
+++ b/vhost-device-gpu/src/backend/virgl.rs
@@ -909,6 +909,8 @@ mod virgl_cov_tests {
             }
             assert!(call_b.read().is_err(), "no signal when no match");
 
+            let capsets = GpuCapset::VIRGL | GpuCapset::VIRGL2;
+
             // Initialize virgl ONCE in this forked process; exercise adapter paths
             let cfg = GpuConfigBuilder::default()
                 .set_gpu_mode(GpuMode::VirglRenderer)
@@ -1029,7 +1031,7 @@ mod virgl_cov_tests {
             assert_matches!(gpu.flush_resource(0, dirty), Ok(GpuResponse::OkNoData));
 
             // Test capset queries
-            for index in [0, 1, 3] {
+            for index in 0..capsets.num_capsets() {
                 test_capset_operations(&gpu, index);
             }