Bugfix, squash this later

Signed-off-by: Ludvig Liljenberg <[email protected]>

Author: ludfjig

src/hyperlight_host/src/mem/mgr.rs

@@ -337,8 +337,12 @@ where
             None => {
                 log_then_return!("Snapshot manager not initialized");
             }
-            Some(snapshot_manager) => snapshot_manager
-                .pop_and_restore_state_from_snapshot(&mut self.shared_mem, dirty_bitmap),
+            Some(snapshot_manager) => {
+                let old_rgns = self.mapped_rgns;
+                self.mapped_rgns = snapshot_manager
+                    .pop_and_restore_state_from_snapshot(&mut self.shared_mem, dirty_bitmap)?;
+                Ok(old_rgns - self.mapped_rgns)
+            }
         }
     }