diff --git a/cloud-hypervisor/tests/integration.rs b/cloud-hypervisor/tests/integration.rs
index b09d72cf9..592759739 100644
--- a/cloud-hypervisor/tests/integration.rs
+++ b/cloud-hypervisor/tests/integration.rs
@@ -164,10 +164,7 @@ mod common_parallel {
 
     #[test]
     fn test_cpu_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(2);
+        let guest = basic_regular_guest!(JAMMY_IMAGE_NAME).with_cpu(2);
         _test_cpu_affinity(&guest);
     }