diff --git a/src/cli.rs b/src/cli.rs
index f7c6f92..617b30b 100644
--- a/src/cli.rs
+++ b/src/cli.rs
@@ -155,7 +155,7 @@ pub fn handle_command() {
             if source.is_none() {
                 source = Some("HEAD".to_string());
             }
-            restore(path, source, worktree, staged);
+            restore(path, source.unwrap(), worktree, staged);
         }
     }
 }