diff --git a/steam_chat_logger.py b/steam_chat_logger.py
index 82c38d9..26bab68 100755
--- a/steam_chat_logger.py
+++ b/steam_chat_logger.py
@@ -75,6 +75,9 @@ except (wa.LoginIncorrect, FileNotFoundError) as exp:
                 code = plaintext_lines[plaintext_lines.index('Login Code') + 1]
                 break
         
+        Mailbox.close()
+        Mailbox.logout()
+        
         print('code found:', code)
         user.login(email_code=code, language='english')
 
@@ -176,8 +179,3 @@ while True:
         parse_trs(trs)
     else:
         break
-
-# CLOSE
-
-Mailbox.close()
-Mailbox.logout()