diff --git a/setup.py b/setup.py
index a4d2fe1..58703a6 100644
--- a/setup.py
+++ b/setup.py
@@ -83,7 +83,7 @@
     # pip to create the appropriate form of executable for the target platform.
     entry_points={
         'console_scripts': [
-            'tibudecrypt.py=tibudecrypt:_main',
+            'tibudecrypt=tibudecrypt:run',
         ],
     },
 )
diff --git a/tibudecrypt.py b/tibudecrypt.py
index 62245b2..04ce10e 100755
--- a/tibudecrypt.py
+++ b/tibudecrypt.py
@@ -255,6 +255,9 @@ def main(args):
     print(SUCCESS_MESSAGE.format(decrypted_filename=decrypted_filename))
 
 
-if __name__ == '__main__':
+def run():
     ARGS = docopt.docopt(__doc__, version=VERSION)
     sys.exit(main(ARGS))
+
+if __name__ == '__main__':
+    run()