diff --git a/cloud-hypervisor/tests/integration.rs b/cloud-hypervisor/tests/integration.rs
index 3d8c3a891..82f9c5453 100644
--- a/cloud-hypervisor/tests/integration.rs
+++ b/cloud-hypervisor/tests/integration.rs
@@ -243,8 +243,7 @@ mod common_parallel {
 
     #[test]
     fn test_power_button() {
-        let disk_config = UbuntuDiskConfig::new(JAMMY_IMAGE_NAME.to_string());
-        let guest = GuestFactory::new_regular_guest_factory().create_guest(Box::new(disk_config));
+        let guest = basic_regular_guest!(JAMMY_IMAGE_NAME);
         _test_power_button(&guest);
     }