Change connection_class to transport_class