Fix & re-enable check_connection

Should increase test coverage!