diff --git a/sshm/__init__.py b/sshm/__init__.py
index cf437db..fd1d198 100644
--- a/sshm/__init__.py
+++ b/sshm/__init__.py
@@ -3,7 +3,7 @@
 This module can be used to ssh into multiple servers at once.
 
 Example:
-    from sshm import sshm
+    from sshm.lib import sshm
 
     for result in sshm('example[5,8].com', 'ps aux | wc -l'):
             print(result)
diff --git a/sshm/main.py b/sshm/main.py
index 9303e08..72f474a 100644
--- a/sshm/main.py
+++ b/sshm/main.py
@@ -10,7 +10,7 @@
 except ImportError: # pragma: no cover version specific
     from sshm.lib import sshm
 
-__all__ = ['sshm']
+__all__ = ['main']
 
 
 def get_argparse_args(args=None):