diff --git a/setup.py b/setup.py
index 0dccca1..9da738a 100644
--- a/setup.py
+++ b/setup.py
@@ -30,7 +30,7 @@ setup(
     zip_safe=False,
     platforms='any',
     entry_points={
-        'console_scripts': ['wakatime = wakatime.__init__:main'],
+        'console_scripts': ['wakatime = wakatime.__init__:execute'],
     },
     classifiers=(
         'Development Status :: 5 - Production/Stable',