From f63fd2d005aca8b940c056894b50230cce1b5287 Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Wed, 18 Feb 2015 10:37:48 +0530
Subject: [PATCH] ...

---
 src/calibre/utils/winreg/default_programs.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/calibre/utils/winreg/default_programs.py b/src/calibre/utils/winreg/default_programs.py
index 4c3e37ac2a..79186cd678 100644
--- a/src/calibre/utils/winreg/default_programs.py
+++ b/src/calibre/utils/winreg/default_programs.py
@@ -163,4 +163,6 @@ class Register(Thread):
 
 if __name__ == '__main__':
     del sys.path[0]
-    unregister()
+    func = (register if sys.argv[-1] == 'register' else unregister)
+    print ('Running', func.__name__, '...')
+    func()