1 files changed, 6 insertions, 3 deletions

diff --git a/src/main.rs b/src/main.rs
index 7393af9..487d095 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -46,13 +46,16 @@ fn main() -> color_eyre::Result<()> {
             let backup = Backup::create(&config, package_manager)?;
             backup.save(&config)?;
         }
-        Subcommands::Restore => { 
+        Subcommands::Restore { package_install } => {
             let Some(last_backup) = Backup::get_last(&config)? else {
                 return Err(Error::BackupNotFound)?;
             };
 
-            last_backup.packages.install()?;
-            last_backup.restore();
+            if package_install {
+                last_backup.packages.install()?;
+            }
+
+            last_backup.restore(&config)?;
         }
     };
     Ok(())