Fix Pep8 in test_connection