diff --git a/src/extension.ts b/src/extension.ts
index e8f5788..992824f 100644
--- a/src/extension.ts
+++ b/src/extension.ts
@@ -377,7 +377,7 @@ export function activate(context: vscode.ExtensionContext) {
           "Cancel"
         );
 
-        if (confirm === "Cancel") {
+        if (confirm !== "Directory and Branch" && confirm !== "Directory Only") {
           return;
         }