diff --git a/vmm/src/lib.rs b/vmm/src/lib.rs
index cf28a38b5..e96b6e3fa 100644
--- a/vmm/src/lib.rs
+++ b/vmm/src/lib.rs
@@ -1209,9 +1209,6 @@ impl Vmm {
 
             // Send last batch of dirty pages
             Self::vm_maybe_send_dirty_pages(vm, &mut socket)?;
-
-            // Stop logging dirty pages
-            vm.stop_dirty_log()?;
         }
         // Capture snapshot and send it
         let vm_snapshot = vm.snapshot()?;
@@ -1231,6 +1228,11 @@ impl Vmm {
             MigratableError::MigrateSend(anyhow!("Error completing migration")),
         )?;
 
+        // Stop logging dirty pages
+        if !send_data_migration.local {
+            vm.stop_dirty_log()?;
+        }
+
         info!("Migration complete");
 
         // Let every Migratable object know about the migration being complete