diff --git a/companion.py b/companion.py
index 961d46e..12e8044 100644
--- a/companion.py
+++ b/companion.py
@@ -2,12 +2,12 @@ from edmc import companion as c
 import platform
 import easygui
 
-companion_session = companion.Session()
+companion_session = c.Session()
 
 def login(username, password):
   try:
     companion_session.login(username, password)
-  except companion.VerificationRequired:
+  except c.VerificationRequired:
     if platform.system() == 'Windows':
       code = easygui.enterbox("Input Verification Code (check your email)",
                               "Verification Required")