More messing around with test_connection.py