diff --git a/setup.py b/setup.py
index db360e6..ef53069 100644
--- a/setup.py
+++ b/setup.py
@@ -24,6 +24,6 @@ setup(
         'Topic :: Communications',
     ],
     entry_points={'console_scripts': [
-        'trezor-agent = sshagent.__main__:main'
+        'trezor-agent = sshagent.__main__:trezor_agent'
     ]},
 )
diff --git a/sshagent/__main__.py b/sshagent/__main__.py
index a39afe1..eb3acb1 100644
--- a/sshagent/__main__.py
+++ b/sshagent/__main__.py
@@ -56,7 +56,7 @@ def parse_args():
     return p.parse_args()
 
 
-def main():
+def trezor_agent():
     args = parse_args()
 
     fmt = ('%(asctime)s %(levelname)-12s %(message)-100s '