diff --git a/cloud-hypervisor/tests/integration.rs b/cloud-hypervisor/tests/integration.rs
index b5c8f0477..d56c25f78 100644
--- a/cloud-hypervisor/tests/integration.rs
+++ b/cloud-hypervisor/tests/integration.rs
@@ -2730,13 +2730,11 @@ mod common_parallel {
 
     #[test]
     fn test_virtio_vsock_hotplug() {
-        let disk_config = UbuntuDiskConfig::new(JAMMY_IMAGE_NAME.to_string());
         #[cfg(target_arch = "x86_64")]
-        let guest = GuestFactory::new_regular_guest_factory().create_guest(Box::new(disk_config));
+        let guest = basic_regular_guest!(JAMMY_IMAGE_NAME);
         #[cfg(target_arch = "aarch64")]
-        let guest = GuestFactory::new_regular_guest_factory()
-            .create_guest(Box::new(disk_config))
-            .with_kernel_path(edk2_path().to_str().unwrap());
+        let guest =
+            basic_regular_guest!(JAMMY_IMAGE_NAME).with_kernel_path(edk2_path().to_str().unwrap());
         _test_virtio_vsock(&guest, true);
     }