From 588997e3127dca13c8899f2ec08c2ff2f19070bc Mon Sep 17 00:00:00 2001
From: Anna Wiggins <annabunches@gmail.com>
Date: Thu, 21 Jan 2016 14:44:23 -0500
Subject: [PATCH] Fix migration errors.

---
 companion.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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")