diff --git a/cloud-hypervisor/tests/integration.rs b/cloud-hypervisor/tests/integration.rs
index 592759739..3d8c3a891 100644
--- a/cloud-hypervisor/tests/integration.rs
+++ b/cloud-hypervisor/tests/integration.rs
@@ -170,10 +170,7 @@ mod common_parallel {
 
     #[test]
     fn test_virtio_queue_affinity() {
-        let disk_config = UbuntuDiskConfig::new(JAMMY_IMAGE_NAME.to_string());
-        let guest = GuestFactory::new_regular_guest_factory()
-            .create_guest(Box::new(disk_config))
-            .with_cpu(4);
+        let guest = basic_regular_guest!(JAMMY_IMAGE_NAME).with_cpu(4);
         _test_virtio_queue_affinity(&guest);
     }