1 files changed, 6 insertions, 4 deletions
diff --git a/e2e_tests/tests/vsock.rs b/e2e_tests/tests/vsock.rs
index 64154289d..dfdb13c00 100644
--- a/e2e_tests/tests/vsock.rs
+++ b/e2e_tests/tests/vsock.rs
@@ -89,7 +89,7 @@ fn host_to_guest_snapshot_restore() {
         ])
         .with_stdout_hardware("legacy-virtio-console");
     drop(vm);
-    vm = TestVm::new_cold_restore(config).unwrap();
+    vm = TestVm::new_restore(config).unwrap();
     host_to_guest_connection(&mut vm, guest_cid, guest_port);
 }
 
@@ -119,7 +119,7 @@ fn host_to_guest_disable_sandbox_snapshot_restore() {
         ])
         .with_stdout_hardware("legacy-virtio-console");
     drop(vm);
-    vm = TestVm::new_cold_restore(config.disable_sandbox()).unwrap();
+    vm = TestVm::new_restore(config.disable_sandbox()).unwrap();
     host_to_guest_connection(&mut vm, guest_cid, guest_port);
 }
 
@@ -199,7 +199,7 @@ fn guest_to_host_snapshot_restore() {
         ])
         .with_stdout_hardware("legacy-virtio-console");
     drop(vm);
-    vm = TestVm::new_cold_restore(config).unwrap();
+    vm = TestVm::new_restore(config).unwrap();
     guest_to_host_connection(&mut vm, host_port);
 }
 
@@ -230,7 +230,7 @@ fn guest_to_host_disable_sandbox_snapshot_restore() {
         ])
         .with_stdout_hardware("legacy-virtio-console");
     drop(vm);
-    vm = TestVm::new_cold_restore(config.disable_sandbox()).unwrap();
+    vm = TestVm::new_restore(config.disable_sandbox()).unwrap();
     guest_to_host_connection(&mut vm, host_port);
 }
 
@@ -277,6 +277,7 @@ fn create_vu_config(cmd_type: CmdType, socket: &Path, cid: u32) -> VuConfig {
 }
 
 #[test]
+#[ignore = "b/333090069 test is flaky"]
 fn vhost_user_host_to_guest() {
     let guest_port = generate_vhost_port();
     let guest_cid = generate_guest_cid();
@@ -295,6 +296,7 @@ fn vhost_user_host_to_guest() {
 }
 
 #[test]
+#[ignore = "b/333090069 test is flaky"]
 fn vhost_user_host_to_guest_with_devices() {
     let guest_port = generate_vhost_port();
     let guest_cid = generate_guest_cid();