diff --git a/aws_google_auth/tests/test_python_version.py b/aws_google_auth/tests/test_python_version.py
index <HASH>..<HASH> 100644
--- a/aws_google_auth/tests/test_python_version.py
+++ b/aws_google_auth/tests/test_python_version.py
@@ -12,17 +12,16 @@ import mock
 
 class TestPythonFailOnVersion(unittest.TestCase):
 
-    @mock.patch('sys.stdout', new_callable=StringIO)
-    def test_python26(self, mock_stdout):
+    def test_python26(self):
 
         with mock.patch.object(sys, 'version_info') as v_info:
             v_info.major = 2
             v_info.minor = 6
 
-            with self.assertRaises(SystemExit):
+            with self.assertRaises(SystemExit) as cm:
                 exit_if_unsupported_python()
 
-            self.assertIn("aws-google-auth requires Python 2.7 or higher.", mock_stdout.getvalue())
+            self.assertEqual(cm.exception.code, 1)
 
     def test_python27(self):
         with mock.patch.object(sys, 'version_info') as v_info:
Optimize test_python<I> test based on log change

In adding the logging functionality, I modified the method
exit_if_unsupported_python so that instead of writing to stdout via
print, it wrote out to logging.critical. This broke the unit test
test_python<I>, as it was written to assert for equality of the first
line of that log output. I adjusted it so that it asserts on the exit
code of 1 instead.