diff --git a/setup.py b/setup.py
index 1eca54a7..69eb959e 100644
--- a/setup.py
+++ b/setup.py
@@ -12,7 +12,7 @@
     entry_points={
         'console_scripts': [
             'atlas=atlasview.atlasview:main',
-            'align=atlaselectrophysiology.ephys_atlas_gui:main',
+            'align=atlaselectrophysiology.ephys_atlas_gui:__main__',
         ]
     },
 )