1 files changed, 4 insertions, 1 deletions
diff --git a/CopyMailto/main.m b/CopyMailto/main.m
index 8dfe2c1..e61f04e 100644
--- a/CopyMailto/main.m
+++ b/CopyMailto/main.m
@@ -11,6 +11,9 @@
 #import "DefaultURLHandler.h"
 
 int main(int argc, const char * argv[]) {
-    [DefaultURLHandler saveDefaultURLHandler];
+    if (argc == 2 && strcmp(argv[1], "--set-url-handler") == 0) {
+        return [DefaultURLHandler setDefaultURLHandler];
+    }
+
     return NSApplicationMain(argc, argv);
 }